是什么原因导致.loc[]返回不一致的形状

Question

我有一个数据帧，它的形状是

print(trainData.shape)
(4146, 17)

的原因是什么

c =  trainData.loc[0,:]
print(c.shape)

返回

(17,)

我需要的输出是一个数据帧，而不是系列

[[0]返回(1,17(

Answer 1

如果您想选择第一行作为数据帧，可以使用

c =  trainData.loc[[0],:]
# or
c =  trainData.loc[0,:].to_frame().T

如果您想选择第一列作为数据帧，

c =  trainData.iloc[:,[0]]
# or
c =  trainData.iloc[:,0].to_frame()