Eu sou novo no Scala e quero calcular uma soma móvel com uma janela fixa para uma lista.
Por exemplo: Dados os valores da lista (1.0, 2.0, 3.0, 6.0, 7.0, 8.0, 12.0, 9.0, 4.0, 1.0) e o período 4, a função deve retornar: (1.0, 3.0, 6.0, 12.0, 18.0, 24,0, 33,0, 36,0, 33,0, 26,0)
Se list.size <period, basta retornar a soma acumulada.
Eu fiz algumas tentativas
def mavg(values: List[Double], period: Int): List[Double] = {
if (values.size <= period) (values.sum ) :: List.fill(period -1)(values.sum ) else {
val rest: List[Double] = mavg(values.tail, period)
(rest.head + ((values.head - values(period)))):: rest
}
}
No entanto, eu tenho
List(12.0, 18.0, 24.0, 33.0, 36.0, 33.0, 26.0, 26.0, 26.0, 26.0
o que não está correto. Não quero usar o Pyspark para obter os resultados. Alguém pode ajudar?
Muito Obrigado.
sliding
método