Почему возникает ошибка в коде?

Question

[Данил Самодуров]

Здравствуйте. В поисках ускорения вычисления экспоненциального скользящего среднего решил написать под это рекурсивный алгоритм. Люблю рекурсию и стараюсь чаще ее применять.
Вот, собственно код:

def get_ema(array, N: int, prev_ema: float = None):
    alpha = 2 / (1 + N)
    if len(array) == 1:
        ema = []
        ema.append(array[0] * alpha + (1 - alpha) * prev_ema)
        return ema

    if not prev_ema:
        for_sma = array[:N]
        sma = sum(for_sma) / len(for_sma)
        return get_ema(array[N:], N, prev_ema=sma).append(sma)
    else:
        new_ema = array[0] * alpha + (1 - alpha) * prev_ema
        return get_ema(array[1:], N, prev_ema=new_ema).append(new_ema)

При запуске возникает ошибка AttributeError: 'NoneType' object has no attribute 'append', указывающая на последнюю строку. Мне не понятно почему. Вроде должен вернуться список, значит append работать должен. Где я ошибся?

Answer 1

[soremix]

append() возвращает None. Так что любой return (кроме return ema) вернет None. Ну и на следующей итерации код посыпется.

Можно быстро проверить, заменив первый блок if

tmp = get_ema(array[N:], N, prev_ema=sma)
print(type(tmp))
return tmp.append(sma)

и второй

tmp = get_ema(array[1:], N, prev_ema=new_ema)
print(type(tmp))
return tmp.append(new_ema)

Comments

[Данил Самодуров]

А можете еще раз объяснить, а то что-то я не понял. Мне казалось, что когда рекурсия развернется, на месте get_ema(array[1:], N, prev_ema=new_ema) уже будет тип list().

[soremix]

Данил Самодуров, не будет.

Возвращается результат вызова метода .append(), а он всегда будет None.

1. На первой итерации выполнится первый блок if.

return get_ema(array[N:], N, prev_ema=sma).append(sma)

2. Он вызовет get_ema(array[N:], N, prev_ema=sma). Во время этого вызова выполнится второй блок if (else). Этот второй блок вызовет функцию еще раз (

get_ema(array[1:], N, prev_ema=new_ema).append(new_ema)

), и вызовет метод append(), который выдаст None.
3. Этот None будет подставлен вместо функции из первого шага (get_ema(array[N:], N, prev_ema=sma)) и в результате в шаге 1 получится None.append(sma)
Итого, функция умрет на 3 итерации

[Данил Самодуров]

soremix, все теперь понял. А как решить проблему в этом случае, если мне нужен массив со всеми скользящими, причем его нужно получить как можно быстрее.

[soremix]

Данил Самодуров,

tmp = get_ema(array[1:], N, prev_ema=new_ema)
tmp.append(new_ema)
return tmp

ну и со вторым аналогично

[Данил Самодуров]

soremix, Точняк. Круто, благодарю.