写一个均线策略的代码

下面是一个均线策略的代码：# 计算简单移动平均线
def SMA(values, n):
return np.convolve(values, np.ones(n), 'valid') / n# 计算指数移动平均线
def EMA(values, n):
weights = np.exp(np.linspace(-1., 0., n))
weights /= weights.sum()
a = np.convolve(values, weights, mode='full')[:len(values)]
a[:n] = a[n]
return a