python - 当在应用中也计算前一个值时，Pandas 中是否可以使用 dataframe.apply 中的前一行值？

Question

我有以下数据框:

Index_Date    A   B     C    D
================================
2015-01-31    10   10   Nan   10
2015-02-01     2    3   Nan   22 
2015-02-02    10   60   Nan  280
2015-02-03    10  100   Nan  250

要求:

Index_Date    A   B    C     D
================================
2015-01-31    10   10    10   10
2015-02-01     2    3    23   22
2015-02-02    10   60   290  280
2015-02-03    10  100  3000  250

Column C 是通过取 D 的 value 为 2015-01-31 派生的。

那我需要将C的value用于2015-01-31并乘以value A 在 2015-02-01 并添加 B。

我尝试了一个 apply 和一个 shift 使用一个 if else 这给出了一个关键错误。

Answer 1

首先，创建派生值:

df.loc[0, 'C'] = df.loc[0, 'D']

然后遍历剩余的行并填充计算值:

for i in range(1, len(df)):
    df.loc[i, 'C'] = df.loc[i-1, 'C'] * df.loc[i, 'A'] + df.loc[i, 'B']


  Index_Date   A   B    C    D
0 2015-01-31  10  10   10   10
1 2015-02-01   2   3   23   22
2 2015-02-02  10  60  290  280