python - 在 Pandas 中提取数据框的第一行和最后一行

Question

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

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

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

但这不会产生 Pandas 数据框:

a    1
b    a
a    4
b    d
dtype: object

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