CSV DictReader를 사용하여 데이터 파일을 쉽게 읽는 방법

[

CSV DictReader를 사용하여 Python에서 CSV 파일 읽기

Python은 csv 모듈을 사용하여 CSV(Comma Separated Values) 파일을 읽고 쓰는 작업을 할 수 있습니다. csv 모듈에는 csv.Reader()와 csv.DictReader()라는 두 가지 방법이 있습니다. csv.Reader()는 인덱스를 사용하여 CSV 데이터에 액세스할 수 있으며, 간단한 CSV 파일에 적합합니다. 반면에 csv.DictReader()는 보다 사용자 친화적이고 쉽게 사용할 수 있으며, 큰 CSV 파일을 다룰 때 유용합니다.

아래는 두 가지 메서드를 사용하여 CSV 파일을 읽는 예제 코드입니다:

csv.Reader() 사용 예제:

import csv

with open('employee_birthday.csv') as csv_file:
    csv_reader = csv.Reader(csv_file, delimiter=',')
    line_count = 0

    for row in csv_reader:
        print(f'\t{row[0]} works in the {row[1]} department, and was born in {row[2]}')

csv.DictReader() 사용 예제:

import csv

with open('employee_birthday.csv') as csv_file:
    csv_reader = csv.DictReader(csv_file, delimiter=',')
    line_count = 0

    for row in csv_reader:
        print(f'\t{row["name"]} works in the {row["department"]} department, and was born in {row["month"]}')

위의 코드에서는 csv.Reader()를 사용하여 CSV 파일을 읽는 방법과 csv.DictReader()를 사용하여 CSV 파일을 읽는 방법을 보여주고 있습니다. 첫 번째 예제에서는 인덱스를 사용하여 각 열의 데이터에 액세스하고 있고, 두 번째 예제에서는 헤더 이름을 사용하여 데이터에 액세스하고 있습니다.

위의 예제를 통해 Python에서 CSV 파일을 읽는 방법을 알 수 있습니다. CSV 파일이 커지거나 열을 추가 또는 제거해야 하는 경우 인덱싱 작업이 복잡해질 수 있습니다. 이럴 때 csv 모듈의 csv.DictReader를 사용하면 보다 간편하고 편리하게 작업할 수 있습니다.