python Pandas DataFrame 取單行數據保持dataframe格式的方法

python pandas的一個使用技巧。
Pandas DataFrame 直接取一行數據會得到Series,
如果想讓結果保持原dataframe格式,在索引外加 [] 即可。
示例如下:

import pandas as pd
test_df = pd.DataFrame([[0,'a','b'],[1,'c','d'],[2,'e','f']],columns=['c1','c2','c3'])
>>>test_df
>>>Out[1]: 
   c1 c2 c3
0   0  a  b
1   1  c  d
2   2  e  f
>>>s1 = test_df.loc[0,:]
>>>s1
>>>Out[2]: 
c1    0
c2    a
c3    b
Name: 0, dtype: object
>>>s2 = test_df.loc[[0],:]
>>>s2
>>>Out[3]: 
   c1 c2 c3
0   0  a  b
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章