能否在Pandas中无需循环实现指定行起的动态计算？

原始数据 #

原始的Pandas DataFrame如下：

month     a     b
1         1     2
2         2     2
3         3     4
4         4     3
5         5     6
6         6     2

需求描述 #

从month=3开始，将a列更新为前一行的a值 × 当前行的b值，最终得到目标结果：

month     a     b
    1         1     2
    2         2     2
    3         8     4
    4         24    3
    5         144   6
    6         288   2

解决方案：无需for循环实现 #

完全可以不用for循环，利用Pandas的向量化运算和累计乘积（cumprod）实现，具体步骤如下：

代码实现 #

import pandas as pd

# 构造原始DataFrame
df = pd.DataFrame({
    'month': [1,2,3,4,5,6],
    'a': [1,2,3,4,5,6],
    'b': [2,2,4,3,6,2]
})

# 标记需要更新的行
mask = df['month'] >= 3

# 计算更新后的a值：第二行的a值 × 从第三行开始的b列累计乘积
df.loc[mask, 'a'] = df.loc[1, 'a'] * df.loc[mask, 'b'].cumprod()

print(df)

运行后输出结果与目标完全一致：

month    a  b
0      1    1  2
1      2    2  2
2      3    8  4
3      4   24  3
4      5  144  6
5      6  288  2

原理说明 #

• cumprod()会对month≥3范围内的b列计算累计乘积：第三行b=4，第四行是4×3=12，第五行是12×6=72，第六行是72×2=144
• 用第二行的a值（2）分别乘以这些累计乘积，就得到了对应行的动态更新值，完美匹配需求中的递推逻辑

内容的提问来源于stack exchange，提问作者I eat toast