如何对大熊猫中的多行使用apply（）函数

Question

最近我得到了一个数据集，该数据集从第1天到第k天都有连续数据，现在我正尝试使用Pandas从中创建一个新数据集，应用大小为n + 1的时间窗口（第一个n数据行以'x'表示，第_n + 1_th行数据以'y'表示从数据集的顶部到底部滚动。

我尝试使用apply（）函数，但一次只获取一行数据，因为出于我的问题，我还希望将接下来的n行数据作为输入，因此我编写了一个for循环，但是这很耗时。

n = 3

x = []
y = []
for idx in devices.index:
    x.append(devices.iloc[idx:idx + n, 4:49].values)
    y.append(devices.iloc[idx:idx + n + 1, 'ALARM'].values)
    if idx + n + 1 == devices.index[-1]:
        break

我想知道是否存在替代for循环的替代方法？ 感谢您的帮助！