獲取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'
]
|
以上就是本文的全部內容，希望對大家的學習有所幫助，也希望大家多多支持腳本之家。