在熊猫中提取数据框的第一行和最后一行

Question

如何将给定数据框的第一行和最后一行提取为熊猫中的新数据框？

我尝试使用 iloc 选择所需的行，然后 concat 如下所示：

 df=pd.DataFrame({'a':range(1,5), 'b':['a','b','c','d']})
pd.concat([df.iloc[0,:], df.iloc[-1,:]])

但这不会产生熊猫数据框：

 a    1
b    a
a    4
b    d
dtype: object

原文由 Bryan P 发布，翻译遵循 CC BY-SA 4.0 许可协议

Answer 1

我认为最简单的方法是 .iloc[[0, -1]] 。

 df = pd.DataFrame({'a':range(1,5), 'b':['a','b','c','d']})
df2 = df.iloc[[0, -1]]

print(df2)

   a  b
0  1  a
3  4  d

原文由 su79eu7k 发布，翻译遵循 CC BY-SA 4.0 许可协议

Answer 2

您还可以使用 head 和 tail ：

 In [29]: pd.concat([df.head(1), df.tail(1)])
Out[29]:
   a  b
0  1  a
3  4  d

原文由 Colonel Beauvel 发布，翻译遵循 CC BY-SA 3.0 许可协议