量化交易学习（四十七）backtrader自定义k线形态指标

发表于 2024-04-04 分类于量化

前几天看《量价分析》的时候，书里介绍了一个动态画趋势线的方法链接：。。。。

其中里面介绍了画趋势线依赖的高位支点和低位支点的概念（如下图）。我就想着用backtrader实现这种蜡烛图形态指标，在一番试验后终于搞定了。

指标代码如下：

import backtrader as bt

class Pivot(bt.Indicator):
    lines = ('high_pivot','low_pivot')
    
    plotlines = dict(
        high_pivot=dict(color='green', ls='',marker='d',markersize='7.0'),
        low_pivot=dict(color='red', ls='',marker='d',markersize='7.0')
    )

    # 设置画在主图上
    plotinfo = dict()
    plotinfo['subplot'] = False
    plotinfo['plotlinelabels'] = True

    def __init__(self):
        super(Pivot, self).__init__()
        linehigh = bt.If(bt.And(bt.And(
            self.data.high(-1)>self.data.high(-2),self.data.high(-1)>self.data.high(0)),
            bt.And(self.data.low(-1)>self.data.low(-2),self.data.low(-1)>self.data.low(0))
        ),self.data.high(-1)*1.01,-100)
        # 因为支点是三根k线中间的那条线，所以要往前挪一位
        self.l.high_pivot=linehigh(1)
        
        linelow = bt.If(bt.And(bt.And(
            self.data.high(-1)<self.data.high(-2),self.data.high(-1)<self.data.high(0)),
            bt.And(self.data.low(-1)<self.data.low(-2),self.data.low(-1)<self.data.low(0))
        ),self.data.high(-1)*1.01,-100)
        self.l.low_pivot=linelow(1)