为每个 pandas 行创建一个包含接下来 10 个行列值的向量

Question

我正在尝试从 pandas 列中创建一个包含前 10 个值的向量，并将其作为单元格中的列表重新插入到 pandas 数据框中。

以下代码有效，但我需要为超过 3000 万行的数据帧执行此操作，因此循环执行此操作需要很长时间。

有人可以帮我将其转换为我可以应用的 numpy 函数吗？我还希望能够在 groupby 中应用此功能。

import pandas as pd

df = pd.DataFrame(list(range(1,20)),columns = ['A'])

df.insert(0,'Vector','')
df['Vector'] = df['Vector'].astype(object)

for index, row in df.iterrows():
     df['Vector'].iloc[index] = list(df['A'].iloc[(index-10):index])

我尝试了多种方式，但都无法让它发挥作用。任何帮助将不胜感激。

Answer 1

IIUC

df['New']=[df.A.tolist()[max(0,x-10):x] for x in range(len(df))]
df
Out[123]: 
     A                                      New
0    1                                       []
1    2                                      [1]
2    3                                   [1, 2]
3    4                                [1, 2, 3]
4    5                             [1, 2, 3, 4]
5    6                          [1, 2, 3, 4, 5]
6    7                       [1, 2, 3, 4, 5, 6]
7    8                    [1, 2, 3, 4, 5, 6, 7]
8    9                 [1, 2, 3, 4, 5, 6, 7, 8]
9   10              [1, 2, 3, 4, 5, 6, 7, 8, 9]
10  11          [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
11  12         [2, 3, 4, 5, 6, 7, 8, 9, 10, 11]
12  13        [3, 4, 5, 6, 7, 8, 9, 10, 11, 12]
13  14       [4, 5, 6, 7, 8, 9, 10, 11, 12, 13]
14  15      [5, 6, 7, 8, 9, 10, 11, 12, 13, 14]
15  16     [6, 7, 8, 9, 10, 11, 12, 13, 14, 15]
16  17    [7, 8, 9, 10, 11, 12, 13, 14, 15, 16]
17  18   [8, 9, 10, 11, 12, 13, 14, 15, 16, 17]
18  19  [9, 10, 11, 12, 13, 14, 15, 16, 17, 18]