Pandas Dataframe groupby 重叠

Question

我正在使用 pandas 数据框来读取包含用于机器学习的数据点的 csv。我试图想出一种方法，让我可以索引一个数据框，它会在其中获取该索引和接下来的 N 行。我不想将数据框分组到没有重叠的箱中（即索引 0:4、4:8 等）我想要的是得到这样的结果：索引 0:4、1:5、 2:6等这将如何完成？

Answer 1

也许您可以创建一个 DataFrame 列表，例如：

import pandas as pd
import numpy as np

nrows = 7
group_size = 5

df = pd.DataFrame({'col1': np.random.randint(0, 10, nrows)})
print(df)

grp = [df.iloc[x:x+5,] for x in range(df.shape[0] - group_size + 1)]
print(grp[1])

原始数据框：

   col1
0     2
1     6
2     6
3     5
4     3
5     3
6     8

DataFrames 列表中的第二个 DataFrame：

   col1
1     6
2     6
3     5
4     3
5     3