获取DataFrame的列名虽然是一个比较简单的操作,但是有时候到手边就是写不出来代码,所以在这里总结记录一下,希望对大家的学习有所帮助。

1.链表推倒式

我们有这样一个数据集:

data_demo=pd.read_csv('dataset.csv')
data_demo=data_demo.iloc[:10,:5]
data_demo

print([column for column in data])

2.通过columns字段获取,返回一个numpy型的array

print(data_demo.columns.values)


注意:这里为一个numpy数组的形式。

3. 直接使用 list 关键字,返回一个list

list(data_demo)

4.df.columns 返回Index,可以通过 tolist(), 或者 list(array) 转换为list

data_demo.columns.tolist()


data.columns 返回的是一个索引对象<class pandas.core.indexes.base.Index'>

Logo

华为开发者空间,是为全球开发者打造的专属开发空间,汇聚了华为优质开发资源及工具,致力于让每一位开发者拥有一台云主机,基于华为根生态开发、创新。

更多推荐