Pandas DataFrame에서 원하는 개수의 행만 확인하고 싶다면 head() 함수를 사용하면 됩니다. 이 함수는 매개변수로 가져올 행의 개수를 지정할 수 있습니다. 예를 들어 상위 10개 행을 조회하고 싶다면 다음과 같이 작성합니다.
dataFrame.head(10)
참고로 매개변수를 지정하지 않으면 기본값으로 상위 5개 행이 반환됩니다.
1. 라이브러리 불러오기
먼저 pandas 라이브러리를 별칭(alias)과 함께 임포트합니다.
import pandas as pd
2. CSV 파일 준비
이 예제에서 사용할 CSV 파일은 데스크톱에 저장되어 있으며, 파일 경로는 다음과 같습니다.
C:\Users\amit_\Desktop\CarRecords.csv
3. DataFrame 생성하기
read_csv() 함수로 CSV 파일을 읽어 Pandas DataFrame을 생성합니다.
dataFrame = pd.read_csv("C:\Users\amit_\Desktop\CarRecords.csv")4. 특정 개수의 행 반환하기
head() 함수에 숫자를 전달하면 해당 개수만큼의 행을 반환합니다. 여기서는 상위 5개 행 레코드를 가져옵니다.
dataFrame.head(5)
전체 예제 코드
지금까지의 내용을 하나로 정리한 전체 코드는 다음과 같습니다.
import pandas as pd
# CSV 파일 읽기
dataFrame = pd.read_csv("C:\Users\amit_\Desktop\CarRecords.csv")
print("DataFrame...\n", dataFrame)
# 상위 5개 행 레코드 반환
print("특정 개수의 행을 가진 DataFrame...\n", dataFrame.head(5))실행 결과
위 코드를 실행하면 다음과 같은 출력 결과를 얻을 수 있습니다.
DataFrame...
Car Place UnitsSold
0 Audi Bangalore 80
1 Porsche Mumbai 110
2 RollsRoyce Pune 100
3 BMW Delhi 95
4 Mercedes Hyderabad 80
5 Lamborghini Chandigarh 80
6 Audi Mumbai 100
7 Mercedes Pune 120
8 Lamborghini Delhi 100
특정 개수의 행을 가진 DataFrame ...
Car Place UnitsSold
0 Audi Bangalore 80
1 Porsche Mumbai 110
2 RollsRoyce Pune 100
3 BMW Delhi 95
4 Mercedes Hyderabad 80마무리 및 팁
head() 함수는 대용량 데이터를 다룰 때 데이터의 구조와 형식을 빠르게 파악하는 데 유용합니다. 반대로 데이터프레임의 마지막 부분을 확인하고 싶다면 tail() 함수를 사용하면 되며, 사용법은 head()와 동일하게 행의 개수를 매개변수로 전달하면 됩니다.