Question

为什么当我切片只包含1行的pandas数据帧时，切片会变成熊猫系列？我怎样才能保留数据框？

df=pd.DataFrame(data=[[1,2,3]],columns=['a','b','c'])
df
Out[37]: 
   a  b  c
0  1  2  3


a=df.iloc[0]

a
Out[39]: 
a    1
b    2
c    3
Name: 0, dtype: int64

Answer 1

要避免重新转换回DataFrame的中间步骤，请在建立索引时使用双括号：

a = df.iloc[[0]]
print(a)
   a  b  c
0  1  2  3

速度：

%timeit df.iloc[[0]]
192 µs per loop

%timeit df.loc[0].to_frame().T
468 µs per loop

Answer 2

使用to_frame()和T进行转置：

df.loc[0].to_frame()

   0
a  1
b  2
c  3

和

df.loc[0].to_frame().T

   a  b  c
0  1  2  3

OR

选项＃2使用双括号[[]]

df.iloc[[0]]

   a  b  c
0  1  2  3

Answer 3

或者您可以按索引切片

a=df.iloc[df.index==0]

a
Out[1782]: 
   a  b  c
0  1  2  3

当切片1行pandas数据帧时，切片变为一系列

3 个答案: