Pandas 데이터프레임의 열 헤더 목록 얻기

소개

Pandas 라이브러리는 데이터 분석을 위한 파이썬의 중요한 도구 중 하나입니다. 이 글에서는 Pandas 데이터프레임의 열(컬럼) 헤더를 리스트 형태로 얻는 방법을 다룹니다. 데이터프레임의 열 헤더는 열의 이름을 나타내며, 때로는 이를 리스트로 추출해야 할 경우가 있습니다.

코드 오류 이름

None

단계별 가이드

데이터프레임 생성

먼저 Pandas를 사용하여 간단한 데이터프레임을 만듭니다. 아래는 예시 코드입니다.

import pandas as pd
data = {'Name': ['Alice', 'Bob', 'Charlie'], 'Age': [25, 30, 35]}
df = pd.DataFrame(data)

columns 속성 사용

Pandas 데이터프레임의 columns 속성을 사용하면 열 헤더를 얻을 수 있습니다. 이 columns 속성은 인덱스 객체를 반환합니다.

column_headers = df.columns

리스트로 변환

인덱스 객체를 리스트로 변환하려면 tolist() 메소드를 사용합니다.

column_headers_list = column_headers.tolist()

예제 코드

이제 지금까지 배운 것을 합쳐 하나의 예제 코드로 만들어 보겠습니다.

import pandas as pd

# 데이터프레임 생성
data = {'Name': ['Alice', 'Bob', 'Charlie'], 'Age': [25, 30, 35]}
df = pd.DataFrame(data)

# 열 헤더 얻기
column_headers = df.columns

# 리스트로 변환
column_headers_list = column_headers.tolist()

# 결과 출력
print(column_headers_list)

이 코드를 실행하면, 출력 결과로 ['Name', 'Age']를 얻을 수 있습니다.

정리

이 글에서는 Pandas 데이터프레임에서 열 헤더를 리스트로 추출하는 방법에 대해 알아보았습니다. columns 속성과 tolist() 메소드를 이용하면 간단하게 이 작업을 수행할 수 있습니다. 이 정보를 활용하여 다양한 데이터 분석 작업을 더 효율적으로 수행하시길 바랍니다.