一、故事由来

我的好朋友燃哥观察了这个指标很久，在元旦以前推荐给我，讨论是否可以转化成量化。 可惜拖延症犯了，一直拖到现在才来帮他完成这样一个心愿，其实也是最近对算法的领悟突飞猛进。 估摸着某一天写一个pine的翻译器。一切皆可python。。 好了废话不多说，我们来介绍一下这个传说中的超级趋势线。。

二、系统介绍

CMC Markets 新一代智能交易系统 —— 超级趋势线（Supertrend） 这里有一篇文章介绍这个系统。

在CMC Markets中的新一代智能交易系统中，在技术指标中选取“超级趋势线”调取即可使用， 如图中所示，可以根据自身喜好对上涨的信号、下跌的信号调节“颜色和粗细”。 那么什么是超趋势指标？在理解超趋势指标公式之前，理解ATR是必要的，因为超趋势使用ATR值来计算指标值。

其中的主要算法下面也有一张图来介绍

大致看一下，主要描述是HL2（k线均价）乘以n倍ATR的通道。做趋势突破。 但文章写得比较简略。没有详细的算法。随后我想到了最牛的社区Tradingview。 果不奇然。上面果然有。

从图上看，还是比较切合趋势的。但可惜的是它只是一个Alert的报警信号。

三、学习源码

看着代码还不算太长，那我们就翻译过来试一下吧。！(っ•̀ω•́)っ✎⁾⁾！ 完整pine代码如上。。

四、代码转化

这里我们在FMZ新建一个策略，起名SuperTrade

接着我们来设置2个参数Factor、Pd

为了更好的简化代码的操作，便于理解，这样要用到python的高级数据扩展包pandas

中午吃饭的时候我问梦梦老师，FMZ是否支持这个库。下午一看居然可以用了。 梦梦老师真的太厉害了。

1.我们要导入pandas库time库 2.在main函数当中设置使用季度合约（主要跑okex） 3.设定一个循环doTicker()15分钟检测1次。 将代码跑在15分钟的周期上 接着我们在doTicker()中写主要策略。

import pandas as pd
import time

def main():
    exchange.SetContractType("quarter")
    preTime = 0
    Log(exchange.GetAccount())
    while True:
        records = exchange.GetRecords(PERIOD_M15)
        if records and records[-2].Time > preTime:
            preTime = records[-2].Time
            doTicker(records[:-1])
        Sleep(1000 *60)
        

4.我们要取回k线的OHCLV 所以用GetRecords（） 5.我们将取回的数据导入pandas M15 = pd.DataFrame(records) 6.我们要修改表的头部标签。 M15.columns = [‘time’,‘open’,‘high’,‘low’,‘close’,‘volume’,‘OpenInterest’] 其实就是将’open’,‘high’,‘low’,'close’ 的首字母改成小写，便于后期写代码不要一会大写一会小写。

def doTicker(records):
    M15 = pd.DataFrame(records)
    M15.columns = ['time','open','high','low','close','volume','OpenInterest']  

7.给数据集合增加一列hl2 hl2=(high+low)/2

#HL2
M15['hl2']=(M15['high']+M15['low'])/2

8.接着我们来计算ATR 因为ATR的计算要导入一个变量length，它的取值是Pd

接着我们通过查阅麦语言手册，ATR真實波動幅度均值的算法步骤如下： TR : MAX(MAX((HIGH-LOW),ABS(REF(CLOSE,1)-HIGH)),ABS(REF(CLOSE,1)-LOW)); ATR : RMA(TR,N)

其中TR的值取下面3个差值的最大一个 1、当前交易日的最高价与最低价间的波幅 HIGH-LOW 2、前一交易日收盘价与当个交易日最高价间的波幅 REF(CLOSE,1)-HIGH) 3、前一交易日收盘价与当个交易日最低价间的波幅 REF(CLOSE,1)-LOW) 所以TR : MAX(MAX((HIGH-LOW),ABS(REF(CLOSE,1)-HIGH)),ABS(REF(CLOSE,1)-LOW));

在python计算中

M15['prev_close']=M15['close'].shift(1)

要先设立一个prev_close 去取close在上一行的数据，也就是将close右移1格成立一个新的参数

ranges= [M15['high'] - M15['low'],M15['high']-M15['prev_close'],M15['low']-M15['prev_close']]

接着定义一个中间变量 记录TR的3个对比值的数组。（HIGH-LOW）（high-prev_close）（low-prev_close）

M15['tr'] = pd.DataFrame(ranges).T.abs().max(axis=1)

我们在数据集合当中定义新的一列取名TR，TR的取值是取中间变量绝对值的最大一个，使用abs（）和max（）函数

    alpha = (1.0 / length) if length > 0 else 0.5
    M15['atr']=M15['tr'].ewm(alpha=alpha, min_periods=length).mean()

最后我们要计算ATR的值，ATR : RMA(TR,N)，据查RMA的算法其实就是一个固定值变种的EMA算法。 N是我们导入的变量，其中ATR的默认参数是14。这里我们导入alpha=length的倒数。

===

然后用ewm算法计算ema 完整ATR计算过程如下

    #ATR(PD)
    length=Pd
    M15['prev_close']=M15['close'].shift(1)
    ranges= [M15['high'] - M15['low'],M15['high']-M15['prev_close'],M15['low']-M15['prev_close']]
    M15['tr'] = pd.DataFrame(ranges).T.abs().max(axis=1)
    alpha = (1.0 / length) if length > 0 else 0.5
    M15['atr']=M15['tr'].ewm(alpha=alpha, min_periods=length).mean()

9始计算Up和Dn

    M15['Up']=M15['hl2']-(Factor*M15['atr'])
    M15['Dn']=M15['hl2']+(Factor*M15['atr'])

Up=hl2 -（Factor * atr） Dn=hl2 +（Factor * atr） 是不是很简单呢。

下面是TV当中15行-21行的核心代码段

TrendUp=close[1]>TrendUp[1]? max(Up,TrendUp[1]) : Up
TrendDown=close[1]<TrendDown[1]? min(Dn,TrendDown[1]) : Dn

Trend = close > TrendDown[1] ? 1: close< TrendUp[1]? -1: nz(Trend[1],1)
Tsl = Trend==1? TrendUp: TrendDown

linecolor = Trend == 1 ? green : red

这一段的主要意思是想表达， 如果处于看涨阶段，（下方线）TrendUp = max(Up,TrendUp[1]) 如果处于下跌阶段，（上方线）TrendDown=min(Dn,TrendDown[1]) 也就是说在一个趋势中，ATR的值一直在使用一种类似强盗布林策略的技术。 不断将通道的另一侧收窄

这里TrendUp和TrendDown每一次的计算都需要进行自我迭代。 就是每一步都要拿上一步的自己来计算。 所以要对数据集合做循环遍历。

这里先要对数据集合新建字段TrendUp，TrendDown，Trend，linecolor。并给定他们一个初始值 接着使用fillna(0)语法将之前计算的结果中带有空值的数据填上0

    M15['TrendUp']=0.0
    M15['TrendDown']=0.0
    M15['Trend']=1
    M15['Tsl']=0.0
    M15['linecolor']='Homily'
    M15 = M15.fillna(0)

启用一个for循环 在循环中采用python三目运算

    for x in range(len(M15)):

计算TrendUp TrendUp = MAX(Up,TrendUp[-1]) if close[-1]>TrendUp[-1] else Up 大致意思是 如果 上一个close>上一个TrendUp，成立取Up和上一个TrendUp当中最大的值，不成立取Up值，并传递给当前TrendUp

        M15['TrendUp'].values[x] = max(M15['Up'].values[x],M15['TrendUp'].values[x-1]) if (M15['close'].values[x-1]>M15['TrendUp'].values[x-1]) else M15['Up'].values[x]

同理，计算TrendDown TrendDown=min(Dn,TrendDown[-1]) if close[-1]<TrendDown[-1] else Dn 大致意思是 如果 上一个close<上一个TrendDown，成立取Dn和上一个TrendDown当中最小的值，不成立取Dn值，并传递给当前TrendDown

        M15['TrendDown'].values[x] = min(M15['Dn'].values[x],M15['TrendDown'].values[x-1]) if (M15['close'].values[x-1]<M15['TrendDown'].values[x-1]) else M15['Dn'].values[x]

下面是计算控制方向的flag，我简化了一下伪代码 Trend= 1 if (close > TrendDown[-1]) else (x) x = -1 if (close< TrendUp[-1]) else Trend[-1]

意义是是 如果 收盘价>上一个 TrendDown 则取1（看多） 不成立取x 如果 收盘价<上一个 TrendUp 则取-1（看空）不成立取上一个Trend （意思是是不变） 翻译成图像语言就是突破上轨转换flag看多，突破下轨转换flag看空，其他时间不变。

        M15['Tsl'].values[x] = M15['TrendUp'].values[x] if  (M15['Trend'].values[x]==1) else M15['TrendDown'].values[x]

计算Tsl和Linecolor Tsl= rendUp if (Trend==1) else TrendDown Tsl 是用来在图像上表示SuperTrend 的值。意思是看多的时候在图上标记下轨，看空的时候在图上标记上轨。 linecolor= ‘green’ if (Trend==1) else ‘red’ linecolor 的含义是 如果看多 则标记绿线 ，如果看空则标记空色（主要是用途Tradingview展示）

        M15['Tsl'].values[x] = M15['TrendUp'].values[x] if  (M15['Trend'].values[x]==1) else M15['TrendDown'].values[x]
        M15['linecolor'].values[x]= 'green' if ( M15['Trend'].values[x]==1) else  'red'

接着23-30行的代码主要是plot绘图 这里不做详解。

最后还有2行代码用于买入卖出信号控制 Tradingview中，他的含义是 反转了Flag以后给出信号 将条件语句转换成为python。 如果上一个Trend flag从-1变成1 代表突破上方阻力 开多 如果上一个Trend flag从1变成-1 代表突破下发支撑 开空

    if(M15['Trend'].values[-1] == 1 and M15['Trend'].values[-2] == -1):
        Log('SuperTrend V.1 Alert Long',"Create Order Buy)
    if(M15['Trend'].values[-1] == -1 and M15['Trend'].values[-2] == 1):
        Log('SuperTrend V.1 Alert Long',"Create Order Sell)

本段完整代码如下：

    M15['TrendUp']=0.0
    M15['TrendDown']=0.0
    M15['Trend']=1
    M15['Tsl']=0.0
    M15['linecolor']='Homily'
    M15 = M15.fillna(0)
    
    for x in range(len(M15)):
        M15['TrendUp'].values[x] = max(M15['Up'].values[x],M15['TrendUp'].values[x-1]) if (M15['close'].values[x-1]>M15['TrendUp'].values[x-1]) else M15['Up'].values[x]
        M15['TrendDown'].values[x] = min(M15['Dn'].values[x],M15['TrendDown'].values[x-1]) if (M15['close'].values[x-1]<M15['TrendDown'].values[x-1]) else M15['Dn'].values[x]
        M15['Trend'].values[x] = 1 if (M15['close'].values[x] > M15['TrendDown'].values[x-1]) else ( -1 if (M15['close'].values[x]< M15['TrendUp'].values[x-1])else M15['Trend'].values[x-1] )
        M15['Tsl'].values[x] = M15['TrendUp'].values[x] if  (M15['Trend'].values[x]==1) else M15['TrendDown'].values[x]
        M15['linecolor'].values[x]= 'green' if ( M15['Trend'].values[x]==1) else  'red'
        
    if(M15['Trend'].values[-1] == 1 and M15['Trend'].values[-2] == -1):
        Log('SuperTrend V.1 Alert Long',"Create Order Buy)
        Log('Tsl=',Tsl)
    if(M15['Trend'].values[-1] == -1 and M15['Trend'].values[-2] == 1):
        Log('SuperTrend V.1 Alert Long',"Create Order Sell)
        Log('Tsl=',Tsl)

五、全部代码

我调整了一下整体的代码结构。 并将做多做空相关下单指令合并到策略中。 下面是完整代码

'''backtest
start: 2019-05-01 00:00:00
end: 2020-04-21 00:00:00
period: 15m
exchanges: [{"eid":"Futures_OKCoin","currency":"BTC_USD"}]
'''

import pandas as pd
import time

def main():
    exchange.SetContractType("quarter")
    preTime = 0
    Log(exchange.GetAccount())
    while True:
        records = exchange.GetRecords(PERIOD_M15)
        if records and records[-2].Time > preTime:
            preTime = records[-2].Time
            doTicker(records[:-1])
        Sleep(1000 *60)

       
def doTicker(records):
    #Log('onTick',exchange.GetTicker())
    M15 = pd.DataFrame(records)

    #Factor=3
    #Pd=7
    
    M15.columns = ['time','open','high','low','close','volume','OpenInterest']  
    
    #HL2
    M15['hl2']=(M15['high']+M15['low'])/2

    #ATR(PD)
    length=Pd
    M15['prev_close']=M15['close'].shift(1)
    ranges= [M15['high'] - M15['low'],M15['high']-M15['prev_close'],M15['low']-M15['prev_close']]
    M15['tr'] = pd.DataFrame(ranges).T.abs().max(axis=1)
    alpha = (1.0 / length) if length > 0 else 0.5
    M15['atr']=M15['tr'].ewm(alpha=alpha, min_periods=length).mean()


    M15['Up']=M15['hl2']-(Factor*M15['atr'])
    M15['Dn']=M15['hl2']+(Factor*M15['atr'])
    
    M15['TrendUp']=0.0
    M15['TrendDown']=0.0
    M15['Trend']=1
    M15['Tsl']=0.0
    M15['linecolor']='Homily'
    M15 = M15.fillna(0)

    for x in range(len(M15)):
        M15['TrendUp'].values[x] = max(M15['Up'].values[x],M15['TrendUp'].values[x-1]) if (M15['close'].values[x-1]>M15['TrendUp'].values[x-1]) else M15['Up'].values[x]
        M15['TrendDown'].values[x] = min(M15['Dn'].values[x],M15['TrendDown'].values[x-1]) if (M15['close'].values[x-1]<M15['TrendDown'].values[x-1]) else M15['Dn'].values[x]
        M15['Trend'].values[x] = 1 if (M15['close'].values[x] > M15['TrendDown'].values[x-1]) else ( -1 if (M15['close'].values[x]< M15['TrendUp'].values[x-1])else M15['Trend'].values[x-1] )
        M15['Tsl'].values[x] = M15['TrendUp'].values[x] if  (M15['Trend'].values[x]==1) else M15['TrendDown'].values[x]
        M15['linecolor'].values[x]= 'Long' if ( M15['Trend'].values[x]==1) else  'Short'
 

    linecolor=M15['linecolor'].values[-2]
    close=M15['close'].values[-2]
    Tsl=M15['Tsl'].values[-2] 


    if(M15['Trend'].values[-1] == 1 and M15['Trend'].values[-2] == -1):

        Log('SuperTrend V.1 Alert Long','Create Order Buy')
        Log('Tsl=',Tsl)
        position = exchange.GetPosition()
        if len(position) > 0:
            Amount=position[0]["Amount"]
            exchange.SetDirection("closesell")
            exchange.Buy(_C(exchange.GetTicker).Sell*1.01, Amount);
        
        exchange.SetDirection("buy")
        exchange.Buy(_C(exchange.GetTicker).Sell*1.01, vol);

    if(M15['Trend'].values[-1] == -1 and M15['Trend'].values[-2] == 1):
        Log('SuperTrend V.1 Alert Long','Create Order Sell')
        Log('Tsl=',Tsl)
        position = exchange.GetPosition()
        if len(position) > 0:
            Amount=position[0]["Amount"]
            exchange.SetDirection("closebuy")
            exchange.Sell(_C(exchange.GetTicker).Buy*0.99,Amount);
        exchange.SetDirection("sell")
        exchange.Sell(_C(exchange.GetTicker).Buy*0.99, vol*2);

公开策略连接https://www.fmz.com/strategy/200625

六、回测与总结

我们选取了近一年的数据进行回测。 使用okex季度合约 15分钟周期。 设定的参数是， Factor=3 Pd=45 vol=100（每次下单100张） 所得年化收益，约33%。 总体来说回撤并不是很大， 其中主要是312的大跌对系统产生了比较大的冲击， 如果没有312的话收益应该会比较好看。

六、写在最后

SuperTrend是一个非常不错的交易系统

SuperTrend系统的主要原理是采用ATR通道突破策略（类似于肯特通道） 但其变化的地方主要在于使用了强盗布林的收窄策略，或者说是逆向的唐奇安原理。 在行情运行中不断收窄上下通道。 以便达到通道突破转向的操作。（一旦通道突破，上下轨恢复初始值）

我在TradingView上把up dn TrendUp TrendDn 分别plot了出来 这样便于更好的理解这个策略 一目了然

另外github上还有一个js的版本。js我不是很懂，但从if语句看好像有点问题。 地址是https://github.com/Dodo33/gekko-supertrend-strategy/blob/master/Supertrend.js

最后我去追查了一下原版。 它发表在2013.05.29 作者是Rajandran R C++代码发表在Mt4论坛https://www.mql5.com/en/code/viewcode/10851/128437/Non_Repainting_SuperTrend.mq4 我大致看懂了C++的意思，有机会再重写一份。

希望大家可以从中学到精髓。 难搞哦。~！