def ab(df):
middle = df[df['c']<df['d']]
return middle['b'].sum() - middle['a'].sum()
result = df.groupby('id').apply(ab)
print(result)
其中df为你的原始数据文件(DataFrame格式),groupby根据id分组,之后每一个分组调用函数ab即可
另:你的原始数据中d列第三个数为20180930,与计算结果不符合,如果按照原始数据计算,结果应为3,3


雷达卡


京公网安备 11010802022788号







