Как сложить значение в столбце v, умножив предыдущую ячейку на фиксированный коэффициент, а затем добавив текущую ячейку ⇐ Python

[Anonymous]

Начиная с df "pd.DataFrame" :~
n v
0 1 0.0
1 2 0.0
2 3 0.0
3 4 0.0
4 5 0.0
5 6 0.0


Я хотел бы сложить значение в столбце «v», при этом ячейка в столбце «v» получается путем умножения предыдущей ячейки «v» на фиксированный коэффициент, а затем сложения текущее значение ячейки столбца «n». (См. пример таблицы расчета ниже)
## sample calculation table :~
n v n + v[i-1] * fixed factor
1 1.0 = 1 + 0.0 * 0.5
2 2.5 = 2 + 1.0 * 0.5
3 4.25 = 3 + 2.5 * 0.5
4 6.125 = 4 + 4.25 * 0.5
5 8.0625 = 5 + 6.125 * 0.5
6 10.03125 = 6 + 8.063 * 0.5

Удалось сделать это с помощью построчной итерации (см. цикл for ниже).
Однако я думаю, что векторизованные методы (такие как cumsum и сдвиг) может быть более эффективным, но не могу понять как; потому что cumsum усложняется умножением, начиная с пустого столбца v, и необходимо ссылаться на предыдущую ячейку столбца.
Интересно, как это сделать с помощью векторизованных методов?Чтобы воспроизвести мой код:
df = pd.DataFrame({'n':[1,2,3,4,5,6]})
df['v'] = 0.0

def fnv(nu, vu):
return nu + vu * 0.5

for i in range(0, df.shape[0]):
df.v.at = fnv(df.n.at, df.v.at[i-1] if i>0 else 0.0)

df (RESULTS) :~
n v
0 1 1.00
1 2 2.50
2 3 4.25
3 4 6.12
4 5 8.06
5 6 10.03


Подробнее здесь: https://stackoverflow.com/questions/790 ... ixed-facto