python中的移动平均线

Question

I've tried to create a generator function which can find simple moving average of the list, but it doesn't work at all. I think the root of the problem might be in finding sum

def gen(lst, n):
    s = 0
    for i, _ in enumerate(lst):
        if lst[i] < n:
            s += lst[n - (n-i)]

my code does nothing when i play it. What should I do?

Answer 1

if lst[i] < n:

You do not define i

but I guess i is a index of lst so try:

for i, _ in enumerate(lst):
  if lst[i] < n:
    s += lst[n - (n-i)]

EDIT:

def gen(lst, n):
    if n > len(lst):
        print("Sorry can't do this")
    try:
        for i, _ in enumerate(lst[:-(n-1)]):
            s = sum(lst[i:i+n])
            yield s/n
    except ZeroDivisionError:
        print("Sorry can't do this")

Answer 2

You may have some issues with this code snippet because you did not include some relevant variables and how they are used.

Below is an example of how you can calculate a moving average of list_of_numbers , given a position p with a period of n .

def mAverage(list_of_numbers, p, n):
    if n < p:
        return sum(list_of_numbers[p-n:p])/n
    else:
        return sum(list_of_numbers[0:p])/(p + 1)

Note: At the beginning of the list this function will only calculate the moving average with a period equal to the position in the list.