python - 如何计算 Python 3 中的移动平均线？

Question

假设我有一个列表:

y = ['1', '2', '3', '4','5','6','7','8','9','10']

我想创建一个函数来计算 n 日移动平均值。 因此，如果 n 为 5，我希望我的代码计算前 1-5，将其相加并求平均值，即 3.0，然后继续计算 2-6，计算平均值，这将是 4.0，然后是 3-7、4-8、5-9、6-10。

我不想计算前 n-1 天，所以从第 n 天开始，它会计算前几天。

def moving_average(x:'list of prices', n):
    for num in range(len(x)+1):
        print(x[num-n:num])

这似乎打印出了我想要的:

[]
[]
[]
[]
[]

['1', '2', '3', '4', '5']

['2', '3', '4', '5', '6']

['3', '4', '5', '6', '7']

['4', '5', '6', '7', '8']

['5', '6', '7', '8', '9']

['6', '7', '8', '9', '10']

但是，我不知道如何计算这些列表中的数字。有什么想法吗？

Answer 1

旧版本的 Python 文档中有一个很棒的滑动窗口生成器 itertools examples :

from itertools import islice

def window(seq, n=2):
    "Returns a sliding window (of width n) over data from the iterable"
    "   s -> (s0,s1,...s[n-1]), (s1,s2,...,sn), ...                   "
    it = iter(seq)
    result = tuple(islice(it, n))
    if len(result) == n:
        yield result    
    for elem in it:
        result = result[1:] + (elem,)
        yield result

使用移动平均线是微不足道的:

from __future__ import division  # For Python 2

def moving_averages(values, size):
    for selection in window(values, size):
        yield sum(selection) / size

针对您的输入运行这个(将字符串映射到整数)给出:

>>> y= ['1', '2', '3', '4','5','6','7','8','9','10']
>>> for avg in moving_averages(map(int, y), 5):
...     print(avg)
... 
3.0
4.0
5.0
6.0
7.0
8.0

要返回“不完整”集合的第一个 n - 1 次迭代，None，只需扩展 moving_averages 函数一点:

def moving_averages(values, size):
    for _ in range(size - 1):
        yield None
    for selection in window(values, size):
        yield sum(selection) / size