nrows, n번째 행까지 불러오기

파이썬에서는 데이터 파일을 불러오기 위해서는 pandas 라이브러리의 read_csv 함수를 이용하면 손쉽게 데이터를 불러올 수 있다.

read_csv 함수는 단순히 파일을 가져오는 것뿐만 아니라 다양한 옵션을 지정해 파일을 불러 올 수 있다.
예를 들어 n번째 행까지 데이터를 불러온다든지 컬럼으로 사용할 행을 지정한다든지 등의 옵션이 있다.

 

데이터의 n번째 행까지 불러오기

데이터를 n번째 행까지만 불러오기 위해서는 read_csv의 nrow 옵션을 이용하면 된다.

예를 들어 5번째 행까지 데이터를 불러오고 싶다면, pd.read_csv('파일경로',nrows=5)로 지정해주면 된다.

# 데이터를 3번째 행까지만 불러오기(nrow=3)

import pandas as pd

train_3 = pd.read_csv('data/train.csv',nrows=3)
test_3 = pd.read_csv('data/test.csv',nrows=3)