准备后数据后,就可以开发构建量化投资策略了。比较知名、流行的量化策略回测框架有vnpy、pyalgotrader、backtrader等。下面以backtrader为例,来运行一个最简单的投资策略。
先安装backtrader的库:pip install backtrader
然后在ChatGPT中输入提示词:
写一段Python代码,用Backtrader库构建一个买入贵州茅台并一直持有的投资策略:
买入资金量:100万元;
仓位:100%
买入时期:2019/1/2
卖出日期:2023/06/13
数据来源:F盘的stocklist_baostock.csv ,贵州茅台股票代码为:sh.600519
stocklist_baostock.csv 的列名称为:date、code、open、high、low、close、preclose、volume、amount、adjustflag、turn、tradestatus、pctChg、isST
运行结果:
Starting Portfolio Value: 1000000.00
Final Portfolio Value: 2879186.44
就是说,2019/1/2买入100万的贵州茅台,2023/06/13卖出,100万就变成了287万,收益率相当不错。
stocklist_baostock.csv是2019/1/2到2023/06/13期间,所有A股股票的日k线数据,一共有467M。这个数据可以参考《零代码量化投资:用ChatGPT下载沪深京 A 股历史行情数据》来自己下载,也可以在“AIGC部落”中下载。