Como calcular a mediana móvel em pandas: com exemplos

Uma mediana móvel é a mediana de vários períodos anteriores em uma série temporal.

Para calcular a mediana móvel de uma coluna em um DataFrame do pandas, podemos usar a seguinte sintaxe:

 #calculate rolling median of previous 3 periods
df[' column_name ']. rolling (3). median ()

O exemplo a seguir mostra como usar esta função na prática.

Exemplo: Calcular a mediana móvel de uma coluna

Suponha que temos o seguinte DataFrame do pandas:

 import pandas as pd

#createDataFrame
df = pd. DataFrame ({' month ': [1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12],
                   ' leads ': [13, 15, 16, 15, 17, 20, 22, 24, 25, 26, 23, 24],
                   ' sales ': [22, 24, 23, 27, 26, 26, 27, 30, 33, 32, 27, 25]})

#view DataFrame
df

	month sales leads
0 1 13 22
1 2 15 24
2 3 16 23
3 4 15 27
4 5 17 26
5 6 20 26
6 7 22 27
7 8 24 30
8 9 25 33
9 10 26 32
10 11 23 27
11 12 24 25

Podemos usar a seguinte sintaxe para criar uma nova coluna contendo a mediana móvel de “vendas” dos 3 períodos anteriores:

 #calculate 3-month rolling median
df[' sales_rolling3 '] = df[' sales ']. rolling (3). median ()

#view updated data frame
df

	month leads sales sales_rolling3
0 1 13 22 NaN
1 2 15 24 NaN
2 3 16 23 23.0
3 4 15 27 24.0
4 5 17 26 26.0
5 6 20 26 26.0
6 7 22 27 26.0
7 8 24 30 27.0
8 9 25 33 30.0
9 10 26 32 32.0
10 11 23 27 32.0
11 12 24 25 27.0

Podemos verificar manualmente se a mediana móvel de vendas exibida para o mês 3 é a mediana dos 3 meses anteriores:

• Mediana de 22, 24, 23 = 23,0

Da mesma forma, podemos verificar a média móvel de vendas do mês 4:

• Mediana de 24, 23, 27 = 24,0

Podemos usar sintaxe semelhante para calcular a mediana móvel de 6 meses:

 #calculate 6-month rolling median
df[' sales_rolling6 '] = df[' sales ']. rolling (6). median ()

#view updated data frame
df

month leads sales sales_rolling3 sales_rolling6
0 1 13 22 NaN NaN
1 2 15 24 NaN NaN
2 3 16 23 23.0 NaN
3 4 15 27 24.0 NaN
4 5 17 26 26.0 NaN
5 6 20 26 26.0 25.0
6 7 22 27 26.0 26.0
7 8 24 30 27.0 26.5
8 9 25 33 30.0 27.0
9 10 26 32 32.0 28.5
10 11 23 27 32.0 28.5
11 12 24 25 27.0 28.5

Recursos adicionais

Os tutoriais a seguir explicam como realizar outras operações comuns em pandas:

Como calcular uma média móvel em Pandas
Como calcular a correlação deslizante em pandas
Como calcular a variação percentual em pandas