选择 pandas 列中的每 n 个值

Question

我想从每 3 个值的列中进行选择。

例如：

  Input
   12 
   73
   56 
   33
   16


                  output
                    12
                    73
                    56 
                    ------
                    73
                    56
                    33
                    -----
                    56
                    33
                    16

我尝试添加一个关键列并按它进行分组，但我的数据框太大而无法执行分组。这是我的尝试：

df.groupby('key').agg(lambda x: x.tolist())

Answer 1

如果使用列表类型，你可以这样做：

lst = [12,73,56,33,16]
slide_size=3
result = []

for i in range(0,len(lst)-slide_size+1):
    result.append(lst[i:i+3])

result

# output : [[12, 73, 56], [73, 56, 33], [56, 33, 16]]

在此之后，您可以将列表转换为DataFrame