在backtrader中，如何自定义指标并使用呢？

Question

本人刚刚解除量化编程。在使用backtrader回测框架时，碰到一个问题。
在init方法中，如何实现当前（K线开盘价-前一根K线开盘价）这个指标呢？
请各位专家教我，感谢

def init(self):
self.order = None
self.buyprice = None
self.buycomm = None
self.dataopen = self.datas[0].open
self.dataclose = self.datas[0].close
self.datalow = self.datas[0].low
self.datahigh = self.datas[0].high
self.datavolume = self.datas[0].volume

Answer 1

在 backtrader 中，你可以使用 LineBuffer 类来创建自定义指标。LineBuffer 是 backtrader 提供的一个数据缓冲区，可以用来存储指标数据或者其他任意数据。

要创建自定义指标，你需要：

在 init 方法中创建 LineBuffer 对象。
在 next 方法中，计算你想要的数据并添加到 LineBuffer 中。
使用 plotinfo 参数在图表中显示你的指标。
例如，你可以这样创建自定义指标：

def init(self):
    self.open_diff = bt.indicators.LineBuffer(self.data.open)


def next(self):
    self.open_diff[0] = self.data.open[0] - self.data.open[-1]

然后，你可以在图表中显示这个指标，例如：

cerebro.plot(y=['open_diff'], ylabel='Open Diff')

基于你提供的代码，我修改了一下，你可以这样实现自定义指标：

def init(self):
    # 创建 LineBuffer 对象
    self.open_diff = bt.indicators.LineBuffer(self.data.open)

    # 使用 plotinfo 参数在图表中显示指标
    self.plotinfo.ploty = True
    self.plotinfo.plotname = 'Open Diff'

def next(self):
    # 计算当前（K线开盘价-前一根K线开盘价）
    self.open_diff[0] = self.data.open[0] - self.data.open[-1]

你可以在 init 方法中使用 plotinfo 参数来定义如何在图表中显示你的指标。在这里，我使用 ploty 参数来启用 y 轴，并使用 plotname 参数来设置指标的名称。

然后，在 next 方法中，我们计算当前 K 线开盘价与前一根 K 线开盘价之差，并将结果存储到 LineBuffer 对象中。