Calculo de juros compostos de um DataFrame Python | Data Science

Bom dia

Estou com duvida de como calcular certas faixas de valores de uma serie de dados. Tenho esses dados abaixo :

             valor
data    
1989-06-01    19.68
1989-07-01    35.90
1989-08-01    36.92
1989-09-01    39.92
1989-10-01    40.64
...    ...
2020-12-01    0.96
2021-01-01    2.58
2021-02-01    2.53
2021-03-01    2.94
2021-04-01    1.51
383 rows × 1 columns

Esses dados são referente ao IGP-M que peguei do site do banco central, valores mensais... Eu gostaria de calcular o acumulado anual e jogar em outra variável. Lembrando que a soma precisa ser feita na base de juros compostos. A saída seria algo desse tipo:


             valor acumulado anual
Ano    
1989        x
1990        x
1991        x
...    ...
2019        x
2020        x

Estou quebrando a cabeça, sei que deve ter alguma função simples, se alguém puder me dar uma dica... Agradeço!

Olá Rafael, tudo bem ? Espero que sim.

Acredito que a função cumsum pode te ajudar com seu objetivo. Ela vai fazer a soma acumulada do seus dados.

Fiz um exemplo com um DataFrame salvo na variável df:

df

	data	valor
0	1989-06-01	19.68
1	1989-07-01	35.9
2	1989-08-01	36.92
3	1989-09-01	39.92
4	1989-10-01	40.64
5	2020-12-01	0.96
6	2021-01-01	2.58
7	2021-02-01	2.53
8	2021-03-01	2.94
9	2021-04-01	1.51

Vamos selecionar a coluna valor e utilizar o cumcum que vai retornar a soma acumulada. (Coloquei também o rename para mudar o nome da coluna resultante.

valor_acumulado_anual = df.valor.cumsum().rename('valor_acumulado_anual')
valor_acumulado_anual

	valor_acumulado_anual
0	19.68
1	55.58
2	92.5
3	132.42
4	173.06
5	174.02
6	176.6
7	179.13
8	182.07
9	183.58

Depois inclui essa coluna no nosso dataframe original.

df['valor_acumulado_anual'] = valor_acumulado_anual
df

	data	valor	valor_acumulado_anual
0	1989-06-01	19.68	19.68
1	1989-07-01	35.9	55.58
2	1989-08-01	36.92	92.5
3	1989-09-01	39.92	132.42
4	1989-10-01	40.64	173.06
5	2020-12-01	0.96	174.02
6	2021-01-01	2.58	176.6
7	2021-02-01	2.53	179.13
8	2021-03-01	2.94	182.07
9	2021-04-01	1.51	183.58

Para conhecer mais sobre o cumsum recomendo a leitura da documentação

Espero ter ajudado, mas qualquer duvida não hesite em perguntar.

Caso este post tenha lhe ajudado, por favor, marcar como solucionado ✓. Bons Estudos!