使用数组 Numpy Python 使用常量和其他变量进行迭代

Question

我正在尝试修改下面的函数，以便它在下面给出预期的输出。对于第一个计算，它类似于100+ 100*-87/100 = 13 和等式NonC_Amount + Int_amount * np.cumprod(PnL / 100)。由于 -87 是 PnL 中的第一个元素，因此对于第二次计算，它将为 13 + 100*-4/100 = 9。 NonC_Amounts 值已更新。

PnL = np.array([-87., -4., -34.1, 8.5])
Int_amount = 100
NonC_Amount = 100
PnL.prod(initial=Int_amount)
NonCompounding =NonC_Amount  + Int_amount  * np.cumprod(PnL / 100)

电流输出：

[ 13, 103.48 , 98.81332,  99.8991322]

预期输出：

[ 13,  9,  -25.1,  -16.6]

Answer 1

你做错了计算。从你的描述看来你想做的事

NonC_Amount = 100
NonCompounding = np.zeros_like(PnL)
for i in range(PnL.shape[0]):
    NonCompounding[i] = NonC_Amount + Int_amount * PnL[i] / 100
    NonC_Amount = NonCompounding[i]

编辑：如果你希望这个操作矢量化，你可以这样做

NonCompounding = NonC_Amount + np.cumsum(Int_amount * PnL / 100)

---

np.cumprod() 为您提供到第 i 个元素的累积乘积。对于np.cumprod(PnL)，那就是

[-87, -87*(-4), -87*(-4)*(-34.1), etc...]

它只是偶然地为您提供第一个元素的正确结果。