数据框中的累积或滚动乘积

Question

您好，我有一个包含一列的数据框，我只想添加另一列，该列采用原始列的滚动乘积。我已经在谷歌上搜索了一段时间，但这似乎是一个基本的功能——不确定我是否遗漏了一些东西。我喜欢将 B 列作为输出。

A   B
1   1
2   2
3   6
4   24
5   120
6   720
7   5040

如果它存在的话，我基本上是在寻找这样的东西：

数据['B'] = 数据['A'].rolling(window=1).product()

我从之前找到了这篇文章，但它似乎正在使用不再活跃的 rolling_apply？：

How to calculate rolling cumulative product on Pandas DataFrame

我在这里尝试过使用类似的解决方案，但它似乎不起作用。

dftest= pd.DataFrame([1,2,3,4,5,6,7],columns=['A'])
dftest['cum']=dftest['A'].rolling(1).apply(lambda x:x.prod())

输出：

   A  cumprod
0  1  1.0
1  2  2.0
2  3  3.0
3  4  4.0
4  5  5.0
5  6  6.0
6  7  7.0

Answer 1

好像你想要cumprod

df = pd.DataFrame({'v':[1,2,3,4,5,6]})
df['prod'] = df.v.cumprod()

    v   prod
0   1   1
1   2   2
2   3   6
3   4   24
4   5   120
5   6   720

---

也可以

df.v.expanding().agg(lambda a:a.prod())

0      1.0
1      2.0
2      6.0
3     24.0
4    120.0
5    720.0