파이썬은 데이터를 파일에 저장하고 읽어오는데 사용되는 파일 객체와 CSV 파일 처리에 매우 유용한 기능을 제공합니다. 이번 블로그 포스트에서는 파이썬에서 파일 객체와 CSV 파일을 다루는 방법을 알아보겠습니다.
파일 객체
파일 객체는 파일을 열고 읽고 쓸 수 있는 파이썬의 내장 클래스입니다. 파일 객체를 사용하여 파일을 열고 데이터를 읽거나 쓸 수 있습니다. 다음은 파일을 열고 읽는 간단한 예제 코드입니다.
with open('파일명.txt', 'r') as file:
data = file.read()
print(data)
위의 코드에서 open()
함수로 파일을 열고 with
문을 사용하여 파일을 자동으로 닫습니다. read()
함수를 사용하여 파일의 내용을 읽고 출력합니다.
파일을 쓰기 모드로 열고 데이터를 파일에 쓰는 예제는 다음과 같습니다.
with open('파일명.txt', 'w') as file:
data = "Hello, World!"
file.write(data)
위의 코드에서는 open()
함수의 두번째 인자로 'w'
를 사용하여 파일을 쓰기 모드로 엽니다. 그리고 write()
함수를 사용하여 데이터를 파일에 씁니다.
CSV 파일 처리
CSV(Comma Separated Values) 파일은 데이터를 쉼표로 구분하여 저장하는 형식입니다. 파이썬은 CSV 파일을 처리하는 데에도 매우 편리한 기능을 제공합니다. CSV 파일을 읽는 예제 코드는 다음과 같습니다.
import csv
with open('data.csv', 'r') as file:
reader = csv.reader(file)
for row in reader:
print(row)
위의 코드에서는 csv
모듈을 사용하여 CSV 파일을 읽습니다. reader()
함수를 사용하여 파일 객체를 CSV 리더 객체로 변환하고, for
루프를 사용하여 각 행을 읽고 출력합니다.
CSV 파일에 데이터를 쓰는 예제 코드는 다음과 같습니다.
import csv
data = [
['Name', 'Age', 'Country'],
['John', '25', 'USA'],
['Alice', '30', 'UK'],
['Bob', '35', 'Canada']
]
with open('data.csv', 'w') as file:
writer = csv.writer(file)
writer.writerows(data)
위의 코드에서는 csv.writer()
함수를 사용하여 CSV 작성기 객체를 가져온 다음, writerows()
함수를 사용하여 전체 데이터를 파일에 작성합니다.
파일 객체와 CSV 파일 처리는 파이썬에서 데이터를 파일로 저장하거나 읽어올 때 매우 유용한 기능입니다. 이러한 기능을 효과적으로 사용하면 데이터를 보다 쉽게 관리하고 처리할 수 있습니다.