【原生代码】Python3 实现ATR、MA、EMA、SMMA、RMA、TEMA指标的计算

1. 参数说明

r：K线数据，字典或者数组
days：指标长度
name：使用哪一个字段，填’Close’即可，如果不填则代表r是数组而不是字典

变量r 字典结构图如下：

{
    {
        'Time': 0,
        'Close': 0,
        'Open': 0,
        'High': 0,
        'Low': 0,
        'Volume': 0,
    }
}

2. ATR 真实波动幅度 （需配合下面的指标）

def ATR(r, days, ma='sma'):
    tr = [0]
    for i in range(len(r)):
        if i:
            tr.append(max(r[i]['High']-r[i]['Low'], abs(r[i]['High'] - r[i-1]['Close']), abs(r[i]['Low'] - r[i-1]['Close'])))

    if ma == 'rma':
        return RMA(tr, days)

    if ma == 'ema':
        return EMA(tr, days)

    return SMA(tr, days)

3. MA 移动平均线

def SMA(r,