python获取Pandas列名的几种方法

 获取DataFrame虽然是一个比较简单的操作,但是有时候到手边就是写不出来,所以在这里总结记录一下:

1.链表推倒式

data = pd.read_csv('data/Receipt code January minute trading volume.csv')

print([column for column in data])

#打印结果

['COUNT', 'SUCC', 'FAIL', 'WAIT PAY', 'SUCCRatio', 'time']

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

print(data.columns.values)

# 打印结果

['COUNT' 'SUCC' 'FAIL' 'WAIT PAY' 'SUCCRatio' 'time']

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

print(list(data))

#打印结果

['COUNT', 'SUCC', 'FAIL', 'WAIT PAY', 'SUCCRatio', 'time']

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

print(data.columns.tolist())

#打印结果

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

['COUNT', 'SUCC', 'FAIL', 'WAIT PAY', 'SUCCRatio', 'time']

以上是 python获取Pandas列名的几种方法 的全部内容, 来源链接: utcz.com/z/346196.html

回到顶部