[TIL] 220804 파이썬 csv 파일 행 단위 읽기, 행 넘기기 next()

csv형식의 파일을 파이썬으로 읽어서 처리할 때

대부분 맨 위에 있는 제목을 넘겨서 next(data)로 필요한 숫자부분만 처리할 수 있었다.

import csv
f = open('age2021.csv', encoding = 'cp949')
data = csv.reader(f)
next(data)
print(data)

 

 그런데 만약 맨 위의 타이틀 부분이 필요하다면, next(data)를 활용해서 한 줄을 타이틀로 지정해 리스트에 넣을 수 있다.

 

import csv
f = open('age2021.csv', encoding = 'cp949')
data = csv.reader(f)
title = next(data)
print(title)

['행정구역', '2021년_거주자_총인구수', '2021년_거주자_0~9세', '2021년_거주자_10~19세']

 

마찬가지로 파일의 두번째 행 부분이 필요하다면, next(data)로 한번 더 넘어가서 읽어온다.

 

import csv
f = open('age2021.csv', encoding = 'cp949')
data = csv.reader(f)
title = next(data) #파일의 맨 윗부분 title에 저장
second_row = next(data) #파일의 두번째 행 second_row에 저장
print(second_row)

['서울특별시  (1100000000)', '9401888 ', '583025 ', '753309 ', '1414854 ']