python里想自己定义一个SMA（移动加权平均）的函数

Posted 2023-03-24

python里想自己写一个SMA（移动加权平均）的函数，我自己定义了SMA（A，n,m,old_sma）,return (m / n) * A + ((n - m) / n) * old_sma, sma0等于前n期A的均值，sma1等于(m / n) * A + ((n - m) / n) * sma0，以此类推，每个sma都和上一期有关系。有没有大神教教我后面怎么写啊啊啊啊啊！！！
例如A[1,2,3,4,5,6,7],m=1,n=3,sma0=(1+2+3)/3=3,sma1=(1/3)*4+(2/3)*sma0=10/3,sma2=(1/3)*5+(2/3)*sma1.......怎么把这个计算的过程实现出来？怎么才可以可以只用改变m，n，A就自己出结果？

python是当下十分火爆的编程语言，尤其在人工智能应用方面。如果有心从事编程方向的工作，最好到专业机构深入学习、多实践，更贴近市场，这样更有利于将来的发展。 参考技术A 有人写过SMA了，借你参考
import numpy as npdef calcSma(data, smaPeriod):
j = next(i for i, x in enumerate(data) if x is not None)
our_range = range(len(data))[j + smaPeriod - 1:]
empty_list = [None] * (j + smaPeriod - 1)
sub_result = [np.mean(data[i - smaPeriod + 1: i + 1]) for i in our_range]

return np.array(empty_list + sub_result) 参考技术B 建议提供一些示例数据，和预期结果，更好说明你的需求。
加权平均的变化还是挺多的追问

就例如A[1,2,3,4,5,6,7],m=1,n=3,sma0=(1+2+3)/3=3,sma1=(1/3)*4+(2/3)*sma0=10/3,sma2=(1/3)*5+(2/3)*sma1.......怎么把这个计算的过程实现出来？怎么才可以只用改变m，n，A就自己出结果？

追答

看不懂，几个参数描述都比较含糊，可能要学会计的才看到懂。
你的例子 ，A像是列表数据，你示例是写的A[1,2,3,4,5,6,7]
又想是要参加计算的元素，比如sma1=(1/3)*4+(2/3) A应该是4

n和m: n是上一期的结束吗，比如你的例子，n=3,指1-3个元素的均值
那m是表示，3后面的M个元素的加权均值吗。所以m=1, 值的是求4的加权均值。

用 Python 实现股票指数移动平均线

移动平均指标用于各种交易策略，以发现价格数据的长期趋势。简单移动平均线策略的一个潜在缺点是它们对所有价格的权重相同，而您可能希望最近的价格占有更大的比重。指数移动平均线 (EMA) 是实现这一目标的一种方法。

下面我们通过代码示例详细介绍 EMA 的实现，并将其与 SMA （简单移动平均）进行比较。

EMA 通过加权乘数赋予最新价格更多权重。这个乘数应用于最后一个价格，因此它比其他数据点占移动平均线更大的部分。EMA 是通过采用最近的价格（我们将其称为“时间 t 的价格”) 减去前一个时间段 （EMA_{t-1}）。此差异由您将 EMA 设置为 (N) 并加回到 EMA_{t-1}的时间段数加权。在数学上，我们可以这样写：

你可能已经注意到上面的等式有一个小问题，它是如何开始的？它参考了最后一个时期的 EMA，所以如果你进行第一次计算，它参考的是什么？这通常可以通过替换简单移动平均线 (SMA) 来初始化计算来延缓，这样您就可以在第一次之后的所有时间段构建 EMA。

让我们通过导入我们的包，用 Python 中的一个简单示例来展示它是如何工作的。

import numpy as np
import pandas as pd
import yfinance as yf
import matplotlib.pyplot as plt

从这里开始，我们将构建两个函数来协同工作并计算我们的指标。第一个函数将是我们上面概述的公式的简单实现：

def _calcEMA(P, last_ema, N):
    return (P - last_ema) * (2 / (N + 1)) + last_ema

第二个函数将计算我们所有数据的 EMA，首先使用 SMA 对其进行初始化，然后迭代我们的数据以使用我们的 SMA 列中的值更新每个后续条目，或者调用我们上面定义的 _calcEMA 函数来处理大于 N的值。

def calcEMA(data, N):
    # Initialize series
    data['SMA_' + str(N)] = data['Close'].rolling(N).mean()
    ema = np.zeros(len(data))
    for i, _row in enumerate(data.iterrows()):
        row = _row[1]
        if i < N:
            ema[i] += row['SMA_' + str(N)]
        else:
            ema[i] += _calcEMA(row['Close'], ema[i-1], N)


    data['EMA_' + str(N)] = ema.copy()
    return data

现在，让我们获取一些数据，看看它是如何工作的。我们将拉出比回测更短的时间段，并比较 EMA 和 SMA 的 10、50 和 100 天。

ticker = 'BABA'
yfObj = yf.Ticker(ticker)
data = yfObj.history(ticker, start='2018-01-01', end='2020-12-31')


N = [10, 50, 100]
_ = [calcEMA(data, n) for n in N]


colors = plt.rcParams['axes.prop_cycle'].by_key()['color']


fig, ax = plt.subplots(figsize=(18, 8))
ax.plot(data['Close'], label='Close')
for i, n in enumerate(N, 1):
    ax.plot(data[f'EMA_{n}'], label=f'EMA-{n}', color=colors[i])
    ax.plot(data[f'SMA_{n}'], label=f'SMA-{n}', color=colors[i], linestyle=':')


ax.legend()
ax.set_title(f'EMA and Closing Price Comparison for {ticker}')
plt.show()

您可以在上图中看到，EMA 比 SMA 对最近的变化更敏感。较短的时间范围也比较长的时间范围更具响应性，较长的时间范围具有可以追溯到几个月或更长时间的价格“记忆”。

所有类型的移动平均线都是滞后指标，这意味着它们只能告诉您价格中已经发生了什么。然而，这并不意味着它们不能用于识别趋势和制定使用一个或多个移动平均指标的策略。如果您有想法，请继续进行测试，看看如何结合 EMA、SMA 和其他值来开发新的盈利交易策略。

扫描本文最下方二维码获取全部完整源码和Jupyter Notebook 文件打包下载。

↓↓长按扫码获取完整源码↓↓