我想从 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)
您也可以简单地使用:
list(my_dataframe)