如何进行个股累加?
由bqk1meln创建,最终由bqk1meln 被浏览 21 用户
问题
在使用cumsum计算成交量累加时,如果有好几支股票,那么cumsum的结果就是所有股票的总成交量。而我想要的是每支股票自己的成交量的类加,请问该如何实现。用groupby试过,没成,应该是没学好,请问该如何实现?还有,如果在可视化中如何实现?是用Python自定义模块吗? 谢谢!
https://bigquant.com/experimentshare/2ed8bbbd66d14ab0bc6dbcc20ecc0919
解答
def get_cumsum(x):
x.sort_values('date',inplace=True)
x['cumsum_a'] = x['amount'].cumsum()
return x
df.groupby('instrument').apply(get_cumsum)
参见上方代码 自定义一个函数即可