Python Pandas 累积乘法

Question

我创建了一个小数据框，我想将前一行乘以 0.5，依此类推。

在:

1
2
3
4

输出:

1
0.5
0.25
0.125

我尝试了以下方法，但效果不佳。它不是累积的，可以说是永久的。

x = pd.DataFrame([1, 2, 3, 4])
y = np.zeros(x.shape)

y[0] = 1
yd = pd.DataFrame(y)

k =  yd.shift(1) * 0.5

print (k)

有什么想法吗？谢谢

---

基于上一期的第二个更复杂的问题。

data['y'] = np.where((data['a']<50) & (data['b']>0), data['initial'], pd.Series(0.99, data['y'].index).cumprod() / 0.99)

我尝试了这段代码，但不起作用。如果前提为真，则调用“初始”，否则继续进行累积乘法。

Answer 1

使用numpy.power

np.power(.5, x - 1)

       0
0  1.000
1  0.500
2  0.250
3  0.125

---

或者如 @DSM pointed out (更直观)

.5 ** (x - 1)

       0
0  1.000
1  0.500
2  0.250
3  0.125

---

另一方面，如果您只想要 .5 的严格连续幂

.5 ** pd.Series(range(len(x)))

0    1.000
1    0.500
2    0.250
3    0.125
dtype: float64

使用cumprod的另一种替代方案

pd.Series(.5, x.index).cumprod() / .5

0    1.000
1    0.500
2    0.250
3    0.125
dtype: float64

或者

pd.Series({**dict.fromkeys(range(4), .5), **{0: 1}}).cumprod()

0    1.000
1    0.500
2    0.250
3    0.125
dtype: float64