python cumprod分组求和
时间: 2023-10-14 20:27:23 浏览: 88
可以使用 pandas 库中的 groupby 和 cumprod 方法实现分组求和后的累乘。
假设有如下数据:
```python
import pandas as pd
df = pd.DataFrame({
'group': ['A', 'A', 'B', 'B'],
'value': [2, 3, 4, 5]
})
```
其中,`group` 列表示分组的标识,`value` 列表示需要累乘的值。
则可以按照 `group` 列进行分组,然后对每个分组中的 `value` 列进行累乘,最后得到每个分组的累乘值:
```python
df['cumprod_sum'] = df.groupby('group')['value'].cumprod()
```
结果如下:
```
group value cumprod_sum
0 A 2 2
1 A 3 6
2 B 4 4
3 B 5 20
```
阅读全文