python - 使用来自另一列的偏移变量在Python中检索 future 值

Question

我试图弄清楚如何使用 Python 中单独行中的偏移变量从 future 日期检索值。例如，我有下面的数据框 df，我想找到一种方法来生成 C 列:

Orig A  Orig B  Desired Column C
54         1           76
76         4           46
14         3           46
35         1           -3
-3         0           -3
46         0           46
64         0           64
93         0           93
72         0           72

非常感谢任何帮助，谢谢!

Answer 1

您可以使用 NumPy 来实现矢量化解决方案:

import numpy as np

idx = np.arange(df.shape[0]) + df['OrigB'].values
df['C'] = df['OrigA'].iloc[idx].values

print(df)

   OrigA  OrigB   C
0     54      1  76
1     76      4  46
2     14      3  46
3     35      1  -3
4     -3      0  -3
5     46      0  46
6     64      0  64
7     93      0  93
8     72      0  72