获取dataframe的columns方法总结。
-
创建dataframe
df = pd.DataFrame([[1, 2, 3]], columns=list("ABC"))
结果如下：
A B C
0 1 2 3
- 最常用的方法
col = df.columns # 获取到的col是<class 'pandas.core.indexes.base.Index'>
结果如下：
Index(['A', 'B', 'C'], dtype='object')
这种方法获取的结果可以观察到列名和数据类型，但是无法直接对得到的结果进行操作。
- 获取数组类型的结果
col = df.columns.values # 返回结果是数组
结果如下：
['A' 'B' 'C']
- 获取list类型的结果
col = df.columns.values.tolist() # 方法1
col = df.columns.tolist() # 方法2
col = [column for column in df] # 方法3
col = list(df.columns.values) # 方法4
col = list(df) # 方法5
col = list(df.columns) # 方法6
上面六种方法得到的结果都是
['A', 'B', 'C']
参考：
https://blog.csdn.net/th_num/article/details/80296254