基于两列百分比值迭代计算新列的Pandas实现问题

原代码问题 #

原代码的问题出在逻辑顺序错了：

• 你先算了a列的累积乘积再乘初始值1000，最后才统一乘(1+b)，但实际计算逻辑是每一步的结果都依赖上一步的c值，得依次用前一步的结果乘当前行的a百分比和(1+b百分比)，不是先攒完所有a再处理b。

正确实现方式 #

方法1：循环迭代（看着明白，适合新手）

import pandas as pd

df = pd.DataFrame({'a': [10, 20, 50], 'b': [5, -2, 20]})
current_val = 1000  # 初始值
c_values = []

for idx, row in df.iterrows():
    current_val = current_val * (row['a'] / 100) * (1 + row['b'] / 100)
    c_values.append(current_val)

df['c'] = c_values
print(df)

方法2：用itertools.accumulate（效率更高，适合大数据量）

import pandas as pd
from itertools import accumulate

df = pd.DataFrame({'a': [10, 20, 50], 'b': [5, -2, 20]})
# 先算出每一步的计算因子：(a的百分比) * (1 + b的百分比)
factors = (df['a'] / 100) * (1 + df['b'] / 100)
# 从初始值1000开始，逐步累积相乘
df['c'] = list(accumulate(factors, lambda x, y: x * y, initial=1000))[1:]
print(df)

方法3：纯pandas操作（符合数据分析习惯）

import pandas as pd

df = pd.DataFrame({'a': [10, 20, 50], 'b': [5, -2, 20]})
# 计算每一步的乘积因子
df['factor'] = (df['a'] / 100) * (1 + df['b'] / 100)
# 用累积乘积乘初始值1000得到c列
df['c'] = 1000 * df['factor'].cumprod()
# 去掉临时的factor列，得到目标结果
print(df.drop('factor', axis=1))

最终输出 #

运行任意一种方法，都会得到你要的结果：

a   b      c
0  10   5  105.000
1  20  -2   20.580
2  50  20   12.348

内容的提问来源于stack exchange，提问作者AmirX