Iterando através de DataFrames com considerações de valor anteriores usando apply()

Em pandas, a função apply() é comumente empregada para aplicar uma função a cada linha de um DataFrame . No entanto, surgem desafios quando o valor da linha anterior também é calculado usando o mesmo método apply().

Considere o seguinte DataFrame:

O objetivo é derivar a coluna C:

• Para 31/01/2015, defina-o com o valor de D.
• Para linhas subsequentes, multiplique o valor da linha anterior de C pelo valor da linha atual de A e adicione-o ao valor da linha atual de B.

Solução:

Para conseguir isso, primeiro definimos o valor inicial de C para 31/01/2015:

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

Em seguida, iteramos pelas linhas restantes e atualizamos os valores C com os cálculos desejados:

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

O DataFrame final após estas operações: