Pandas DataFrame 열 전체 리스트 가져오기

Stack Overflow에 자주 검색, 등록되는 문제들과 제가 개발 중 찾아 본 문제들 중에서 나중에도 찾아 볼 것 같은 문제들을 정리하고 있습니다.

Stack Overflow에서 가장 먼저 확인하게 되는 가장 높은 점수를 받은 Solution과 현 시점에 도움이 될 수 있는 가장 최근에 업데이트(최소 점수 확보)된 Solution을 각각 정리하였습니다.

 

아래 word cloud를 통해 이번 포스팅의 주요 키워드를 미리 확인하세요.

Get a list from Pandas DataFrame column headers

Pandas DataFrame 열 전체 리스트 가져오기

 문제 내용 

I want to get a list of the column headers from a Pandas DataFrame. The DataFrame will come from user input, so I won't know how many columns there will be or what they will be called.

Pandas DataFrame에서 열 헤더 리스트를 가져오려고 합니다. 데이터 프레임은 사용자 입력에서 나오기 때문에 몇 개의 열이 있을지, 어떤 이름으로 불릴지 알 수 없습니다.

 

For example, if I'm given a DataFrame like this:

예를 들어, 다음과 같은 데이터 프레임이 제공되는 경우:

>>> my_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

 

I would get a list like this:

나는 다음과 같은 목록을 얻을 것이다:

>>> header_list
['y', 'gdp', 'cap']

 

 

 높은 점수를 받은 Solution 

You can get the values as a list by doing:

다음을 수행하여 값을 리스트로 가져올 수 있습니다.

list(my_dataframe.columns.values)

 

Also you can simply use (as shown in Ed Chum's answer):

또한 Ed Chum의 답변에 나와 있듯이 다음을 사용할 수도 있습니다.

list(my_dataframe)

 

 

 가장 최근 달린 Solution 

Its very simple.

아주 간단해요.

 

Like you can do it as:

다음과 같이 할 수 있습니다.

 

list(df.columns)

 

 

 

출처 : https://stackoverflow.com/questions/19482970/get-a-list-from-pandas-dataframe-column-headers