python - 如何为每周 EMA 编写 CustomFactor - Python、Quantopian

Question

class EWMAWeekly(CustomFactor):

    inputs = [USEquityPricing.close]
    window_length = (13 + 2 * 13 - 1) * 5  # Initial 13 weeks for sma, then 25 more weeks to improve the accuracy of the current ema.

    def compute(
        self,
        today,
        assets,
        out,
        data,
        ):
        alpha = 2 / (13 + 1)
        weekly_data = data[4::5]  # len = 38, index from 0 - 37
        ema = average(weekly_data[:13])  # Initial SMA
        i = 0
        while i < 25:
            ema = weekly_data[13 + i] * alpha + ema * (1 - alpha)
            i += 1

        out[:] = ema

以上CustomFactor是我目前拥有的。当我通过管道运行它时，输出是average(weekly_data[:13])25SMA周前的输出。该代码不会引发任何错误，并且我已经测试了 while 循环，所以我知道它正在运行。我认为问题出ema在 while 循环内重新分配变量。我可能有片刻的愚蠢，但我似乎找不到问题所在。任何建议表示赞赏。

谢谢

Answer 1

看来我犯了一个愚蠢的错误。我使用整数来计算 alpha 而不是浮点数。更正后的代码如下。

class EWMAWeekly(CustomFactor):

inputs = [USEquityPricing.close]
window_length = (13 + 2 * 13 - 1) * 5  # Initial 13 weeks for sma, then 25 more weeks to improve the accuracy of the current ema.

def compute(
    self,
    today,
    assets,
    out,
    data,
    ):
    alpha = 2.0 / (13.0 + 1.0)
    weekly_data = data[4::5]  # len = 38, index from 0 - 37
    ema = average(weekly_data[:13])  # Initial SMA
    i = 0
    while i < 25:
        ema = weekly_data[13 + i] * alpha + ema * (1 - alpha)
        i += 1

    out[:] = ema