获取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'
]
|
以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持脚本之家。