Pandas 指定行の前後を残したい

Question

前提

Pandas で指定行の前後を残すプログラムを書きたいです。
書き方ががわからないためご教授願います。

実現したいこと

下記の表でたべもののカラムのバナナの行を指定して、
バナナの行±２行指定で抽出するプログラムを組みたいです。

対象となる基データ

ID	たべもの	個数
１	リンゴ	１
１	リンゴ	２
１	リンゴ	３
１	リンゴ	４
２	バナナ	０
１	リンゴ	１
１	リンゴ	２
１	リンゴ	３
１	リンゴ	４

コーディングにより出力されるデータ

ID	たべもの	個数
１	リンゴ	３
１	リンゴ	４
２	バナナ	０
１	リンゴ	１
１	リンゴ	２

補足情報（FW/ツールのバージョンなど）

環境
Python 3.9.8

Answer 1

※ テストデータを一部改変しています

import pandas as pd
import io

csv_data = '''
ID,たべもの,個数
1,バナナ,0
1,リンゴ,1
1,リンゴ,2
1,リンゴ,3
1,リンゴ,4
1,リンゴ,5
2,バナナ,0
1,リンゴ,1
1,リンゴ,2
1,リンゴ,3
1,リンゴ,4
1,リンゴ,5
1,リンゴ,6
3,バナナ,5
'''
df = pd.read_csv(io.StringIO(csv_data))

#
idx = [*{*sum([[*range(max(0, i-2), min(i+3, len(df)))]
              for i in df.index[df['たべもの']=='バナナ']], [])}]
dfx = df.iloc[idx]

print(dfx)

ID	たべもの	個数
1	バナナ	0
1	リンゴ	1
1	リンゴ	2
1	リンゴ	4
1	リンゴ	5
2	バナナ	0
1	リンゴ	1
1	リンゴ	2
1	リンゴ	5
1	リンゴ	6
3	バナナ	5

Comments

[watchdogs]

迅速なご回答ありがとうございました。
助かりました。