我想从pandas DataFrame获取列标题的列表。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']
你可以执行以下操作以列表形式获取值:
list(my_dataframe.columns.values)
你也可以简单地使用:(如Ed Chum的答案所示):
list(my_dataframe)