Saada nimekiri pandas DataFrame'i veeru päistest

Ma tahan saada nimekirja veergude päistest pandas DataFrame'ist. DataFrame tuleb kasutaja sisendist, nii et ma'ei tea, mitu veergu seal on või kuidas neid nimetatakse.

Näiteks kui mulle antakse selline DataFrame:

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

Tahaksin saada sellise nimekirja:

>>> header_list
['y', 'gdp', 'cap']
Gulzar
Redigeeritud küsimus T66 juuni 2019 в 11:48
Programmeerimine
python pandas dataframe
T66 oktoober 2013 в 9:18
36 vaated
Lahendus

Saate väärtused loeteluna, tehes:

list(my_dataframe.columns.values)

Samuti võite lihtsalt kasutada: (nagu näidatud Ed Chum'i vastuses):

list(my_dataframe)
cs95
Redigeeritud vastus T22 aprill 2019 в 9:51
Kommentaarid (10)

See on saadaval kui my_dataframe.columns.

BrenBarn
Redigeeritud vastus T33 jaanuar 2014 в 6:50
Kommentaarid (2)
 
n = []
for i in my_dataframe.columns:
    n.append(i)
print n
Kommentaarid (3)