Python 판다스 데이터프레임 열 헤더에서 목록 가져오기, Get a list from Pandas DataFrame column headers

질문

나는 Pandas DataFrame에서 열 헤더 목록을 가져 오려고합니다. DataFrame은 사용자 입력에서 가져올 것이므로 몇 개의 열이 있을지 또는 어떤 이름으로 지정될지 모릅니다.

예를 들어, 다음과 같은 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']

답변

다음과 같이 목록으로 값을 가져올 수 있습니다:

list(my_dataframe.columns.values)

또한, (Ed Chum의 답변에서 보여주는 것처럼) 다음과 같이 간단하게 사용할 수 있습니다:

list(my_dataframe)