파이썬에서 CSV 파일에 데이터를 작성하는 방법

Python에서 CSV 파일에 데이터를 작성하는 단계별 튜토리얼을 제공합니다.

1단계: 필요한 모듈 가져오기 먼저, CSV 파일 작업에 필요한 'csv' 모듈을 가져와야 합니다. 또한 CSV 파일에 작성하려는 데이터를 생성하거나 조작하는 데 필요한 다른 모듈도 가져와야 합니다.

import csv

2단계: 데이터 준비 다음으로, CSV 파일에 작성하려는 데이터를 준비해야 합니다. 이는 리스트, 사전 또는 CSV 형식으로 쉽게 변환할 수 있는 기타 데이터 구조 형식일 수 있습니다. 예를 들어, 직원들을 나타내는 사전의 리스트가 있다고 가정해 보겠습니다.

employees = [
    {"Name": "John Doe", "Age": 30, "Position": "매니저"},
    {"Name": "Jane Smith", "Age": 25, "Position": "어시스턴트"},
    {"Name": "Mike Johnson", "Age": 35, "Position": "감독"}
]

3단계: 쓰기 모드로 CSV 파일 열기 이제 open() 함수를 사용하여 CSV 파일을 쓰기 모드로 열어야 합니다. 파일 경로와 이름을 지정하고, 행 사이에 빈 줄이 삽입되지 않도록 newline='' 매개변수를 사용합니다.

with open('employees.csv', mode='w', newline='') as file:

4단계: CSV 작성자 객체 생성 with 블록 내부에서 csv.writer() 함수를 사용하여 CSV 작성자 객체를 생성합니다. 파일 객체와 구분 기호 문자(기본값은 쉼표)와 같은 다른 선택적 매개변수를 지정할 수 있습니다.

    writer = csv.writer(file)

5단계: 헤더 행 작성 (선택 사항) CSV 파일에 헤더 행을 포함하려면, CSV 작성자 객체의 writerow() 메서드를 사용하여 열 이름의 리스트를 작성할 수 있습니다.

    writer.writerow(["Name", "Age", "Position"])

6단계: 데이터 행 작성 이제 루프를 사용하여 데이터를 반복하고 writerow() 메서드를 사용하여 각 행을 CSV 파일에 작성할 수 있습니다.

    for employee in employees:
        writer.writerow([employee["Name"], employee["Age"], employee["Position"]])

7단계: 파일 닫기 마지막으로, close() 메서드를 사용하여 파일을 닫아 시스템 리소스를 해제해야 합니다.

    file.close()

이것으로 파이썬에서 CSV 파일에 데이터를 성공적으로 작성했습니다. 이제 파일을 열어 CSV 형식으로 저장된 데이터를 확인할 수 있습니다.

전체 코드는 다음과 같습니다.

import csv

employees = [
    {"Name": "John Doe", "Age": 30, "Position": "매니저"},
    {"Name": "Jane Smith", "Age": 25, "Position": "어시스턴트"},
    {"Name": "Mike Johnson", "Age": 35, "Position": "감독"}
]

with open('employees.csv', mode='w', newline='') as file:
    writer = csv.writer(file)
    writer.writerow(["Name", "Age", "Position"])
    for employee in employees:
        writer.writerow([employee["Name"], employee["Age"], employee["Position"]])
    file.close()

이 튜토리얼이 파이썬에서 CSV 파일에 데이터를 작성하는 방법을 이해하는 데 도움이 되기를 바랍니다!