Python Pandas – DataFrame의 모든 열(컬럼) 이름 표시하는 방법

Pandas DataFrame에서 열 이름만 확인하고 싶다면 dataframe.columns 속성을 사용하면 됩니다. 이 속성은 DataFrame이 가진 모든 열의 이름을 Index 객체 형태로 반환합니다.

1. 라이브러리 임포트

먼저 pandas를 별칭(alias)과 함께 임포트합니다. 관례적으로 pd라는 별칭을 사용합니다.

import pandas as pd

2. 예제 DataFrame 생성

자동차 판매 데이터를 담은 간단한 DataFrame을 만들어 보겠습니다. 이 DataFrame은 'Car', 'Place', 'Units' 세 개의 열로 구성되어 있습니다.

dataFrame = pd.DataFrame(
    {
        "Car": ['BMW', 'Audi', 'BMW', 'Lexus', 'Tesla', 'Lexus', 'Mustang'],
        "Place": ['Delhi', 'Bangalore', 'Hyderabad', 'Chandigarh', 'Pune', 'Mumbai', 'Jaipur'],
        "Units": [100, 150, 50, 110, 90, 120, 80]
    }
)

3. 전체 코드

아래는 dataframe.columns를 사용해 모든 열 이름을 출력하는 전체 코드입니다.

import pandas as pd

# DataFrame 생성
dataFrame = pd.DataFrame(
    {
        "Car": ['BMW', 'Audi', 'BMW', 'Lexus', 'Tesla', 'Lexus', 'Mustang'],
        "Place": ['Delhi', 'Bangalore', 'Hyderabad', 'Chandigarh', 'Pune', 'Mumbai', 'Jaipur'],
        "Units": [100, 150, 50, 110, 90, 120, 80]
    }
)

# DataFrame 출력
print("DataFrame ...\n", dataFrame)

# 모든 열 이름 가져오기
print("\n모든 열 이름 표시 ...\n", dataFrame.columns)

4. 실행 결과

위 코드를 실행하면 다음과 같은 결과가 출력됩니다.

DataFrame ...
       Car       Place  Units
0      BMW      Delhi    100
1     Audi  Bangalore    150
2      BMW  Hyderabad     50
3    Lexus Chandigarh    110
4    Tesla       Pune     90
5    Lexus     Mumbai    120
6  Mustang     Jaipur     80

모든 열 이름 표시 ...
Index(['Car', 'Place', 'Units'], dtype='object')

추가 팁: 열 이름을 리스트로 변환하기

.columns의 결과는 Index 객체이므로, 일반 파이썬 리스트로 변환하고 싶다면 list() 함수나 .tolist() 메서드를 사용할 수 있습니다.

column_list = dataFrame.columns.tolist()
print(column_list)   # ['Car', 'Place', 'Units']

또한 열 이름 개수만 필요한 경우에는 len(dataFrame.columns)를 사용하면 간편하게 확인할 수 있습니다.