获取DataFrame列名的3种方法

df= pd.DataFrame({'a': range(10, 20), 'b': range(20, 30)})
df

1.链表推倒式

[column for column in df]
[a,b]

2.通过columns属性

columns属性返回Index，columns.values属性返回 numpy.ndarray，然后可以通过 tolist(), 或者 list（ndarray） 转换为list

print(type(df.columns))
<class 'pandas.core.indexes.base.Index'>
print(type(df.columns.values))
<class 'numpy.ndarray'>
print(type(df.columns.tolist()),":"+"
",df.columns.tolist())
<class 'list'> :
 ['a', 'b']
print(type(df.columns.values.tolist()),":"+"
",df.columns.values.tolist())
<class 'list'> :
 ['a', 'b']

3.直接使用 list ，返回一个含有columnsd的list列表

print(list(df))
['a', 'b']

总结

以上是生活随笔为你收集整理的获取DataFrame列名的3种方法的全部内容，希望文章能够帮你解决所遇到的问题。

如果觉得生活随笔网站内容还不错，欢迎将生活随笔推荐给好友。