在pandas中选择特定行上方和下方的N行

import pandas as pd

# 创建一个示例DataFrame
data = {'A': [1, 2, 3, 4, 5],
        'B': [6, 7, 8, 9, 10],
        'C': [11, 12, 13, 14, 15]}
df = pd.DataFrame(data)

# 选择第3行上方和下方的2行
target_row = 3
N = 2
selected_rows = df.iloc[target_row-N:target_row+N+1]

print(selected_rows)

   A   B   C
1  2   7  12
2  3   8  13
3  4   9  14
4  5  10  15