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