问题描述
你可以执行以下操作以列表形式获取值:
list(my_dataframe.columns.values)
你也可以简单地使用:(如Ed Chum的答案所示):
list(my_dataframe)
解决方法
我想从pandas DataFrame
获取列标题的列表。DataFrame来自用户输入,所以我不知道会有多少列或它们将被称为什么。
例如,如果给我这样的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
我想要一个这样的列表:
>>> header_list
['y','gdp','cap']