我想从 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']
最佳答案
您可以通过执行以下操作获取值作为列表:
list(my_dataframe.columns.values)
您也可以简单地使用(如 Ed Chum's answer 所示):
list(my_dataframe)
关于python - 从 Pandas DataFrame 列标题中获取列表,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/41767704/