pandas rolling_apply cumprod

Question

我正在尝试将滚动累积产品添加到熊猫系列中。我的输入系列是：

   s
0  1
1  2
2  3
3  4
4  5
5  6

我想得到一个结果系列，它给我之前“n”值的累积乘积。所以如果'n'是3，我想得到：

   s
0  n/a
1  n/a
2  6
3  24
4  60
5  120

我提出的代码使用 rolling_apply 和 lambda 函数并产生 TypeError：

import pandas as pnd
df = pnd.DataFrame()
df['s'] = [1,2,3,4]
print (df)
print (pnd.rolling_apply(df.s,2,lambda x : x.cumprod()))

TypeError: only length-1 arrays can be converted to Python scalars

有人知道怎么做吗？

Answer 1

感谢用户 2357112。这是我想出的有效代码...

import pandas as pnd
df = pnd.DataFrame()
df['s'] = [1,2,3,4, 5, 6]
print (df)
print (pnd.rolling_apply(df.s,3,lambda x : x.prod()))