python获取Pandas列名的几种方法

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

1.链表推倒式

1 2 3 4 5 6

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

1 2 3

print(data.columns.values)  # 打印结果 ['COUNT' 'SUCC' 'FAIL' 'WAIT PAY' 'SUCCRatio' 'time']

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

1 2 3

print(list(data)) #打印结果 ['COUNT', 'SUCC', 'FAIL', 'WAIT PAY', 'SUCCRatio', 'time']

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

1 2 3 4

print(data.columns.tolist()) #打印结果 data.columns 返回的是一个索引对象<class 'pandas.core.indexes.base.Index'> ['COUNT', 'SUCC', 'FAIL', 'WAIT PAY', 'SUCCRatio', 'time']

以上就是本文的全部内容，希望对大家的学习有所帮助，也希望大家多多支持脚本之家。