파이썬을 이용한 CSV 파일 다루기

서론

CSV(Comma Separated Values)는 데이터를 저장하는 데 흔히 사용되는 형식입니다. 이 형식은 간단하고 일반적으로 다양한 응용 프로그램에서 지원됩니다. 파이썬은 CSV 파일을 다루기 위한 다양한 기능을 제공하며, 이를 통해 데이터를 읽고 쓰는 작업을 쉽게 할 수 있습니다.

이 포스트에서는 파이썬을 사용하여 CSV 파일을 다루는 방법에 대해 알아보겠습니다.

본론

CSV 파일 읽기

CSV 파일을 읽기 위해서는 csv 모듈을 import 해야 합니다. csv.reader 함수를 사용하여 CSV 파일을 읽을 수 있습니다. 다음은 예시 코드입니다.

“`python
import csv

with open(‘파일명.csv’, ‘r’, newline=”) as file:
reader = csv.reader(file)
for row in reader:
print(row)
“`

위 코드는 파일명.csv 파일을 읽고, 각 줄을 출력합니다. 각 줄은 리스트 형태로 표시되며, 각 항목은 쉼표로 구분됩니다.

CSV 파일 쓰기

CSV 파일을 쓰기 위해서도 csv 모듈을 사용합니다. csv.writer 함수를 사용하여 CSV 파일을 생성하고 데이터를 작성할 수 있습니다. 다음은 예시 코드입니다.

“`python
import csv

data = [
[‘이름’, ‘나이’, ‘성별’],
[‘영희’, 20, ‘여성’],
[‘철수’, 25, ‘남성’],
[‘미영’, 30, ‘여성’]
]

with open(‘파일명.csv’, ‘w’, newline=”) as file:
writer = csv.writer(file)
writer.writerows(data)
“`

위 코드는 파일명.csv 파일을 생성하고, data 리스트의 내용을 작성합니다. writerows 메서드를 사용하여 데이터를 한 번에 작성할 수도 있고, writerow 메서드를 사용하여 한 줄씩 작성할 수도 있습니다.

기타 CSV 파일 다루기 기능

csv 모듈은 CSV 파일을 다루는 데 다양한 기능을 제공합니다. 몇 가지 중요한 기능은 다음과 같습니다.

• delimiter: CSV 파일의 구분 기호를 설정합니다. 기본값은 쉼표입니다.
• quotechar: 인용 부호로 둘러싸인 문자열을 처리하는 데 사용됩니다. 기본값은 큰 따옴표입니다.
• skipinitialspace: True로 설정하면 데이터의 첫 번째 공백을 무시합니다.
• fieldnames: CSV 파일의 첫 번째 줄에 있는 필드 이름을 지정할 수 있습니다. 딕셔너리 형태로 지정할 수도 있습니다.
• dialect: CSV 파일의 특정 방언을 설정할 수 있습니다.

이러한 기능을 잘 활용하여 CSV 파일을 다양한 방식으로 다룰 수 있습니다.

결론

파이썬의 csv 모듈을 사용하면 CSV 파일을 효율적으로 다룰 수 있습니다. CSV 파일을 읽거나 쓰는 기능 외에도 다양한 설정 옵션을 통해 원하는 대로 파일을 처리할 수 있습니다. CSV 파일을 다룰 때는 모듈의 문서를 참조하여 원하는 기능을 적절히 활용하는 것이 중요합니다.