如何在 streamlit 中显示 backtrader 返回的图表？

Question

我尝试使用 python 中的回溯库对股票数据进行回测。 我使用这个简单的策略

class CrossOver(bt.SignalStrategy):
    def __init__(self):
        sma=bt.ind.SMA(period=50)
        price=self.data
        crossover=bt.ind.CrossOver(price,sma)
        self.signal_add(bt.SIGNAL_LONG,crossover)

然后我运行它并尝试 plot 它并以流光显示

cerebro=bt.Cerebro()
cerebro.addstrategy(CrossOver)
cerebro.adddata(data)
cerebro.run()
pl=cerebro.plot()
st.pyplot(pl)

但我无法在流光中看到图表。 有谁知道如何在 streamlit 中显示反向交易者的图表？ 提前致谢。

Answer 1

我对反向交易者不太熟悉，所以我从他们的文档中举了一个例子，说明如何创建 plot。 plot 中使用的数据可以从他们的github 存储库下载。

该解决方案包含以下步骤：

1. 确保我们使用的 matplotlib 后端不会向用户显示绘图，因为我们希望在 Streamlit 应用程序中显示它，并且 backtrader 的 plot() function 显示 Z32FA6E1B78A96D40289Z53E6A60 这可以使用：

    matplotlib.use('Agg')

2. 从 backtrader 的 plot() function 中获取 matplotlib 图。 这可以使用：

    figure = cerebro.plot()[0][0]

3. 以流光显示 plot。 这可以使用：

    st.pyplot(figure)

全部一起：

import streamlit as st
import backtrader as bt
import matplotlib

# Use a backend that doesn't display the plot to the user
# we want only to display inside the Streamlit page
matplotlib.use('Agg')

# --- Code from the backtrader plot example
# data can be found in there github repo 
class St(bt.Strategy):
    def __init__(self):
        self.sma = bt.indicators.SimpleMovingAverage(self.data)
data = bt.feeds.BacktraderCSVData(dataname='2005-2006-day-001.txt')
cerebro = bt.Cerebro()
cerebro.adddata(data)
cerebro.addstrategy(St)
cerebro.run()
figure = cerebro.plot()[0][0]

# show the plot in Streamlit
st.pyplot(figure)

Output：