pandas入门: 获取dataframe的columns

获取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