Python으로 CSV 파일 읽고 쓰는 방법

 

 

Python은 데이터 처리를 위한 강력한 도구 중 하나로, 특히 CSV 파일을 읽고 쓰는 데 매우 유용합니다. CSV는 'Comma Separated Values'의 약자로, 각 데이터 필드가 특정 구분자로 나누어진 텍스트 기반의 파일 형식입니다. 이 글에서는 파이썬을 사용하여 CSV 파일을 어떻게 효율적으로 다룰 수 있는지 알아보도록 하겠습니다.

CSV 파일이란?

 

CSV 파일은 간단한 형식으로 데이터를 저장할 수 있는 기법입니다. 각 줄은 하나의 데이터 레코드를 나타내며, 데이터의 각 필드는 쉼표로 나뉘어져 있습니다. 예를 들어, 다음은 간단한 CSV 파일의 예시입니다:

• Name, Age, City
• John, 25, New York
• Alice, 30, Los Angeles
• Bob, 28, Chicago

위와 같이 CSV 파일은 쉽게 읽고 쓸 수 있으며, 많은 프로그램에서 지원하므로 데이터 공유에 편리합니다.

파이썬의 CSV 모듈 활용하기

파이썬에서는 CSV 파일을 처리하기 위해 'csv'라는 내장 모듈을 제공합니다. 이 모듈을 사용하면 데이터를 손쉽게 읽고 쓸 수 있으며, 다양한 옵션을 설정해 특정 요구사항에 맞게 CSV 파일을 처리할 수 있습니다.

 

CSV 파일 읽기

CSV 파일을 읽기 위해서는 'csv.reader()' 함수를 사용합니다. 이 함수는 파일을 열고 각 행을 리스트 형태로 반환합니다. 예를 들어:

import csv

with open('example.csv', 'r') as file:

reader = csv.reader(file)

for row in reader:

print(row) # 각 행을 출력

위 코드는 CSV 파일의 내용을 행 단위로 읽어 출력합니다.

CSV 파일 쓰기

CSV 파일에 데이터를 작성하려면 'csv.writer()' 함수를 활용합니다. 이 함수는 제공된 리스트 형태의 데이터를 CSV 형식으로 변환하여 파일에 저장합니다. 아래의 코드를 참고하세요:

 

data = [['Name', 'Age', 'City'], ['John', 25, 'New York'], ['Alice', 30, 'Los Angeles']]

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

writer = csv.writer(file)

writer.writerows(data) # 데이터를 한 번에 작성

이 코드는 데이터를 CSV 파일로 안전하게 기록합니다.

CSV 파일 처리의 다양한 옵션

CSV 모듈은 다양한 구분자를 지원합니다. 기본적으로 쉼표가 사용되지만, 탭이나 세미콜론 등 다른 구분자도 지정할 수 있습니다. 이를 통해 각기 다른 형식의 CSV 파일도 효율적으로 처리할 수 있습니다.

 

헤더 처리 및 속성 사용

CSV 파일에서 첫 줄에 필드명이 포함되어 있는 경우, 'csv.DictReader()'를 사용하면 각 행을 딕셔너리 형태로 읽을 수 있어 더 유연한 데이터 처리가 가능합니다. 예를 들면:

with open('example_with_header.csv', 'r') as file:

reader = csv.DictReader(file)

for row in reader:

print(row['Name'], row['Age']) # 필드명으로 접근

특수 문자 처리하기

CSV 파일에 포함된 특수 문자를 효과적으로 다루기 위해 'quoting' 속성을 사용할 수 있습니다. 아래는 quoting 속성의 설정 예시입니다:

• csv.QUOTE_ALL: 모든 필드를 인용 부호로 감쌉니다.
• csv.QUOTE_MINIMAL: 필요한 경우에만 인용 부호를 사용합니다.
• csv.QUOTE_NONNUMERIC: 숫자가 아닌 필드만 인용 부호로 감쌉니다.
• csv.QUOTE_NONE: 완전히 인용 부호를 사용하지 않습니다.

결론

파이썬은 CSV 파일을 다루는 데 매우 유용한 기능을 제공하여 데이터 관리 작업을 간소화합니다. 'csv' 모듈을 활용함으로써 파일을 읽고 쓰는 것은 물론, 다양한 형식과 옵션을 통해 데이터를 유연하게 처리할 수 있습니다. 이를 통해 사용자는 복잡한 데이터 작업을 손쉽게 수행할 수 있습니다.

파이썬을 이용하여 CSV 파일을 처리하는 방법을 잘 이해하고 효과적으로 활용하신다면, 데이터 분석이나 처리 작업이 더욱 매끄럽게 진행될 것입니다. 앞으로 다양한 데이터 작업 시 CSV 파일을 적극적으로 활용해 보시기를 권장합니다.

 

혼자 떠나는 제주도 여행 추천 명소

 

자주 물으시는 질문

CSV 파일이란 무엇인가요?

CSV 파일은 데이터를 쉼표로 구분하여 저장하는 텍스트 형식입니다. 각 줄이 레코드를 나타내며, 다양한 소프트웨어에서 지원하여 데이터 공유에 유용합니다.

Python에서 CSV 파일을 어떻게 읽나요?

Python에서는 'csv.reader()' 함수를 사용하여 CSV 파일을 읽습니다. 이 함수는 파일의 각 행을 리스트 형태로 반환합니다.

CSV 파일에 데이터를 어떻게 작성하나요?

'csv.writer()' 함수를 활용하면 리스트 형태의 데이터를 CSV 형식으로 작성하여 파일에 저장할 수 있습니다. 이를 통해 간편하게 기록할 수 있습니다.

CSV 파일의 헤더를 처리하는 방법은?

CSV 파일의 첫 줄에 필드명이 있는 경우, 'csv.DictReader()'를 사용하면 각 행을 딕셔너리 형태로 읽을 수 있어 데이터 접근이 더 용이해집니다.