from datetime import datetime import backtrader as bt # 升级到最新版 import matplotlib.pyplot as plt # 由于 Backtrader 的问题,此处要求 pip install matplotlib==3.2.2 import akshare as ak # 升级到最新版 import pandas as pd plt.rcParams["font.sans-serif"] = ["SimHei"] plt.rcParams["axes.unicode_minus"] = False # 利用 AKShare 获取股票的后复权数据,这里只获取前 6 列 stock_hfq_df = ak.stock_zh_a_hist(symbol="601916", adjust="hfq").iloc[:, :6] # 处理字段命名,以符合 Backtrader 的要求 stock_hfq_df.columns = [ 'date', 'open', 'close', 'high', 'low', 'volume', ] # 把 date 作为日期索引,以符合 Backtrader 的要求 stock_hfq_df.index = pd.to_datetime(stock_hfq_df['date']) 如何画出数据源的K线图,标示(20,50,200)MA线的代码是
时间: 2024-02-22 13:00:12 浏览: 71
以下是画出数据源的K线图并标示(20,50,200)MA线的代码:
```
# 定义 Backtrader 中的数据源
data = bt.feeds.PandasData(dataname=stock_hfq_df)
# 初始化 Cerebro 对象
cerebro = bt.Cerebro()
# 添加数据源
cerebro.adddata(data)
# 添加 MA 指标
cerebro.addindicator(bt.indicators.SMA, period=20)
cerebro.addindicator(bt.indicators.SMA, period=50)
cerebro.addindicator(bt.indicators.SMA, period=200)
# 添加 K线图和 MA 线图
cerebro.addobserver(bt.observers.Broker)
cerebro.addobserver(bt.observers.Trades)
cerebro.addobserver(bt.observers.DrawDown)
cerebro.addobserver(bt.observers.Value)
cerebro.addobserver(bt.observers.TimeReturn)
cerebro.addanalyzer(bt.analyzers.SharpeRatio)
cerebro.addanalyzer(bt.analyzers.Returns)
cerebro.addanalyzer(bt.analyzers.DrawDown)
cerebro.addanalyzer(bt.analyzers.Transactions)
cerebro.addanalyzer(bt.analyzers.SQN)
cerebro.addanalyzer(bt.analyzers.AnnualReturn)
cerebro.addanalyzer(bt.analyzers.VWR)
# 运行回测
cerebro.run()
# 画出 K 线图和 MA 线图
cerebro.plot(style='candlestick', ma=[20, 50, 200])
```
阅读全文