Computer >> 컴퓨터 >  >> 프로그래밍 >> Python

Pandas에서 헤더(header) 없이 CSV 파일 읽는 방법

Pandas에서 헤더(열 이름)가 포함되지 않은 CSV 파일을 읽으려면 read_csv() 메서드의 header 매개변수를 None으로 설정하면 됩니다.

예제 데이터 준비

이번 예제에서 사용할 CSV 파일(SalesData.csv)의 내용은 다음과 같습니다. 파일을 Microsoft Excel로 열었을 때의 모습입니다.

Pandas에서 헤더(header) 없이 CSV 파일 읽는 방법

1단계: 라이브러리 임포트

먼저 필요한 Pandas 라이브러리를 임포트합니다.

import pandas as pd

2단계: 기본 방식으로 CSV 파일 읽기

CSV 파일의 데이터를 Pandas DataFrame으로 불러오면 첫 번째 행이 자동으로 헤더(열 이름)로 인식됩니다.

dataFrame = pd.read_csv("C:\\Users\\amit_\\Desktop\\SalesData.csv")

3단계: 헤더 없이 CSV 파일 읽기

파일을 불러올 때 header=None 옵션을 지정하면, 첫 번째 행도 일반 데이터로 취급되어 헤더 대신 정수 인덱스(0, 1, 2...)가 열 이름으로 할당됩니다.

pd.read_csv("C:\\Users\\amit_\\Desktop\\SalesData.csv", header=None)

전체 예제 코드

두 가지 방식을 비교할 수 있는 전체 코드는 다음과 같습니다.

import pandas as pd

# CSV 파일을 Pandas DataFrame으로 불러오기 (헤더 포함)
dataFrame = pd.read_csv("C:\\Users\\amit_\\Desktop\\SalesData.csv")
print("\nReading the CSV file...\n", dataFrame)

# CSV 파일을 헤더 없이 불러오기
dataFrame = pd.read_csv("C:\\Users\\amit_\\Desktop\\SalesData.csv", header=None)
print("\nReading the CSV file (without header)...\n", dataFrame)

실행 결과

위 코드를 실행하면 다음과 같은 출력 결과를 확인할 수 있습니다.

Reading the CSV file...
        Car   Reg_Price   Units
0       BMW        2500     100
1     Lexus        3500      80
2      Audi        2500     120
3    Jaguar        2000      70
4   Mustang        2500     110

Reading the CSV file (without header)...
0 1 2
0       Car  Reg_Price Units
1       BMW       2500 100
2     Lexus       3500 80
3      Audi       2500 120
4    Jaguar       2000 70
5   Mustang       2500 110

정리

header=None 옵션을 사용하면 CSV 파일의 첫 번째 행이 열 이름이 아닌 데이터 값으로 처리됩니다. 반대로 CSV 파일에 헤더가 없는 경우에도 이 옵션을 활용하면 데이터가 올바르게 로드됩니다. 추가로 names 매개변수와 함께 사용하면 직접 원하는 열 이름을 지정할 수도 있습니다.