使用來自另一列的偏移量變量在Python中檢索未來值

Question

我試圖弄清楚如何使用Python中單獨行中的偏移量變量從未來日期檢索值。 例如，我有下面的數據幀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

Answer 2

import pandas as pd

dict = {"Orig A": [54,76,14,35,-3,46,64,93,72],
        "Orig B": [1,4,3,1,0,0,0,0,0],
        "Desired Column C": [76,46,46,-3,-3,46,64,93,72]}

df = pd.DataFrame(dict)

df["desired_test"] = [df["Orig A"].values[i+j] for i,j in enumerate(df["Orig B"].values)]

df

   Orig A  Orig B  Desired Column C  desired_test
0      54       1                76            76
1      76       4                46            46
2      14       3                46            46
3      35       1                -3            -3
4      -3       0                -3            -3
5      46       0                46            46
6      64       0                64            64
7      93       0                93            93
8      72       0                72            72