我想從pandas DataFrame獲取列標題的列表。DataFrame來自用戶輸入，所以我不知道會有多少列或它們將被稱為什么。例如，如果給我這樣的DataFrame：>>> my_dataframe y gdp cap0 1 2 51 2 3 92 8 7 23 3 4 74 6 7 75 4 8 36 8 2 87 9 9 108 6 6 49 10 10 7我想要一個這樣的列表：>>> header_list['y', 'gdp', 'cap']
3 回答
ITMISS
TA貢獻1871條經(jīng)驗 獲得超8個贊
您可以執(zhí)行以下操作以列表形式獲取值：
list(my_dataframe.columns.values)
您也可以簡單地使用：（如Ed Chum的答案所示）：
list(my_dataframe)
白板的微信
TA貢獻1883條經(jīng)驗 獲得超3個贊
做了一些快速測試，使用內(nèi)置版本dataframe.columns.values.tolist()最快的也許并不奇怪：
In [1]: %timeit [column for column in df]
1000 loops, best of 3: 81.6 μs per loop
In [2]: %timeit df.columns.values.tolist()
10000 loops, best of 3: 16.1 μs per loop
In [3]: %timeit list(df)
10000 loops, best of 3: 44.9 μs per loop
In [4]: % timeit list(df.columns.values)
10000 loops, best of 3: 38.4 μs per loop
（list(dataframe)盡管我還是很喜歡，所以謝謝EdChum！）
添加回答
舉報
0/150
提交
取消