Pandas DataFrame에서 열 헤더 검색

Pandas DataFrames는 효율적인 데이터 조작 및 분석을 가능하게 하는 다목적 데이터 구조입니다. 일반적인 작업 중 하나는 DataFrame 구조의 개요를 얻거나 추가 처리를 위해 유용할 수 있는 열 헤더를 추출하는 것입니다.

열의 수와 이름을 알 수 없는 사용자 입력에서 가져온 DataFrame이 있다고 가정합니다. . 열 헤더를 목록으로 추출하려면 다음 방법을 활용할 수 있습니다.

방법 1: DataFrame.columns.values ​​사용

column_headers = list(my_dataframe.columns.values)

my_dataframe.columns는 열 헤더가 포함된 Index 객체를 반환합니다. 값을 사용하여 이 인덱스를 목록으로 변환하면 열 이름을 나타내는 문자열 목록을 얻을 수 있습니다.

방법 2: DataFrame.columns 사용

column_headers = list(my_dataframe)

이 메소드는 열을 목록으로 직접 변환하는 my_dataframe.columns.values에 대한 약식 표기법입니다.

사용 예

DataFrame을 고려하세요.

    y  gdp  cap
0   1    2    5
1   2    3    9
2   8    7    2
3   3    4    7
4   6    7    7
5   4    8    3
6   8    2    8
7   9    9   10
8   6    6    4
9  10   10    7

두 방법 중 하나를 사용하면 다음과 같은 열 헤더 목록을 얻을 수 있습니다.

['y', 'gdp', 'cap']