ta.py 文件源码

def add_SMA(self, timeperiod=20,
            type='line', color='secondary', **kwargs):
    """Simple Moving Average."""

    if not self.has_close:
        raise Exception()

    utils.kwargs_check(kwargs, VALID_TA_KWARGS)
    if 'kind' in kwargs:
        type = kwargs['kind']

    name = 'SMA({})'.format(str(timeperiod))
    self.pri[name] = dict(type=type, color=color)
    self.ind[name] = talib.SMA(self.df[self.cl].values,
                               timeperiod)