拒绝踩坑 使用cumsum计算累计求和
首先导入库名
# 导入库名
import pandas as pd
import numpy as np
import random
生成数据样例
grpArr = np.random.choice(['A','B'], size=[1,10])
monArr = np.random.choice(['1月','2月','3月'], size=[1,10])
ordArr = np.random.randint(low=1,high=6, size=[1,10])
salesArr = 60 + 10*np.random.randn(1,10)
data = pd.DataFrame({
'userID':grpArr[0], 'month':monArr[0], 'orders':ordArr[0], 'sales':salesArr[0]})
data.loc[random.randint(0,9), 'orders'] = np.nan
data.loc