从pandas DataFrame列标题获取列表 [英] Get list from pandas DataFrame column headers

问题描述

我想从一个熊猫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 
  pre> 
 
 
我想要得到这样的列表：
  >>> header_list 
 [y，gdp，cap] 
  
 
 
解决方案
您可以通过执行以下操作获取列表值：
 列表（my_dataframe.columns.values）
  
还可以使用：
 列表（my_dataframe）
  
 
I want to get a list of the column headers from a pandas DataFrame.  The DataFrame will come from user input so I won't know how many columns there will be or what they will be called.


For example, if I'm given a DataFrame like this:
>>> 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
I would want to get a list like this:
>>> header_list
[y, gdp, cap]

解决方案
You can get the values as a list by doing:
list(my_dataframe.columns.values)
Also you can simply use:
list(my_dataframe)


                        
这篇关于从pandas DataFrame列标题获取列表的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！