파이썬에서 CSV 파일을 읽는 방법

파이썬에서 CSV 파일을 읽는 방법에 대한 단계별 튜토리얼입니다.

   import csv

이제 읽을 CSV 파일을 열어야 합니다. 이를 위해 open() 함수를 사용할 수 있습니다. 이 함수는 두 개의 인자를 받습니다: 파일 경로와 모드(이 경우 "r"은 읽기를 의미합니다). 예를 들어, CSV 파일의 이름이 "data.csv"이고 파이썬 스크립트와 동일한 디렉토리에 위치한 경우 다음과 같이 열 수 있습니다:

   with open('data.csv', 'r') as file:
       # 파일을 읽는 코드를 여기에 작성합니다.

with open() 문은 파일을 올바르게 닫아주는 역할을 합니다. 파이썬에서 파일을 다룰 때 이 문을 사용하는 것이 좋은 습관입니다.

csv.reader() 함수를 사용하여 CSV 리더 객체를 생성합니다. 이 함수에 파일 객체를 인자로 전달하면 리더 객체를 생성할 수 있습니다. 이 리더 객체를 사용하여 CSV 파일의 각 줄을 순회할 수 있습니다. 다음 코드를 with 블록 안에 추가하세요:

   csv_reader = csv.reader(file)

이제 CSV 파일의 내용을 읽을 수 있습니다. for 루프를 사용하여 파일의 각 줄을 순회할 수 있습니다. 각 줄은 값들의 리스트로 반환됩니다. 이전 라인 아래에 다음 코드를 with 블록에 추가하세요:

   for row in csv_reader:
       # 각 줄을 처리하는 코드를 여기에 작성합니다.

for 루프 안에서 리스트를 인덱싱하여 각 줄의 값을 가져올 수 있습니다. 예를 들어, CSV 파일에 세 개의 열이 있다면 다음과 같이 각 열의 값을 가져올 수 있습니다:

   for row in csv_reader:
       column1 = row[0]
       column2 = row[1]
       column3 = row[2]

인덱싱은 0부터 시작하므로 첫 번째 열은 인덱스 0, 두 번째 열은 인덱스 1이 되는 식입니다.

각 줄의 값을 기반으로 원하는 작업이나 데이터 처리를 수행할 수 있습니다. 예를 들어, 값을 출력하거나 변수에 저장하여 추가 분석에 활용할 수 있습니다. 다음은 각 줄의 값을 출력하는 예시입니다:

   for row in csv_reader:
       print(row)

첫 번째 줄(예: 헤더를 포함하는 경우)을 건너뛰고 싶다면 next() 함수를 for 루프 전에 사용할 수 있습니다. 예를 들어:

   next(csv_reader)  # 첫 번째 줄 건너뛰기
   for row in csv_reader:
       print(row)

마지막으로, 파일을 읽은 후에는 파일을 닫는 것을 잊지 마세요. with open() 문이 자동으로 파일을 닫아주지만 명시적으로 파일을 닫는 것이 좋은 습관입니다. for 루프 이후에 다음 라인을 추가하세요:

   file.close()

여기까지입니다! 이제 파이썬에서 CSV 파일을 읽는 방법을 알았습니다. csv 모듈의 더 고급 기능을 탐색해보세요. 다른 구분 기호나 인용 스타일을 처리하는 방법이나 헤더가 포함된 CSV 파일을 읽는 데 사용되는 DictReader 클래스 등을 알아보세요.