[解決済み] pandasでデータフレームの先頭行と最終行を抽出する

質問

pandasで与えられたデータフレームの最初と最後の行を新しいデータフレームとして抽出するにはどうしたらよいですか？

私は使用しようとしました iloc を使って目的の行を選択し、次に concat のようにします。

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

のようになりますが、これではpandasのdataframeは生成されません。

a    1
b    a
a    4
b    d
dtype: object

どのように解決するのですか？

最もシンプルな方法は .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