Итерация по кадрам данных с учетом предыдущих значений с использованием apply()

В pandas функция apply() обычно используется для применения функции к каждой строке DataFrame . Однако проблемы возникают, когда значение предыдущей строки также рассчитывается с использованием того же метода apply().

Рассмотрим следующий DataFrame:

Цель состоит в том, чтобы получить столбец C:

• На 31 января 2015 г., установите для него значение D.
• Для последующих строк умножьте значение предыдущей строки C на текущее значение строки A и добавьте его к текущему значению строки B.

Решение:

Чтобы добиться этого, мы сначала установили начальное значение C для 31 января 2015 г.:

df.loc[0, 'C'] = df.loc[0, 'D']

Затем мы перебираем оставшиеся строки и обновляем значения C с помощью желаемых вычислений:

for i in range(1, len(df)):
    df.loc[i, 'C'] = df.loc[i-1, 'C'] * df.loc[i, 'A']   df.loc[i, 'B']

Окончательный DataFrame после этих операций: