Atualizando a função definida pelo usuário
Funções definidas pelo usuário são uma ótima forma de reduzir trabalho repetitivo. A função definida no vídeo está abaixo:
def compute_ratio(df, numerator, denominator,
ratio_name):
df[ratio_name] = df[numerator] / df[denominator]
return df
Perceba que a função que definimos não consegue calcular diretamente índices que exigem somar ou subtrair valores no numerador e no denominador, como a margem operacional ou a margem bruta.
Neste exercício, você vai atualizar essa função para que ela possa ser usada para calcular índices que envolvem adição no numerador e no denominador.
Você usará a função do pandas .sum(), que pode somar ao longo de um eixo no DataFrame. Um eixo 0 significa linhas, então somaria os valores ao longo das linhas e retornaria um valor para cada coluna. Um eixo 1 significa colunas, então somaria valores de colunas diferentes em df e retornaria um valor para cada linha. Você pode pensar em df.sum(axis=1) como equivalente a df[column_1] + ... + df[column_n]. O eixo padrão em pd.sum() é 0.
Este exercício faz parte do curso
Análise de Demonstrativos Financeiros em Python
Exercício interativo prático
Experimente este exercício completando este código de exemplo.
# Update the function
def compute_ratio(df, numerator, denominator, ratio_name):
df[ratio_name] = df[numerator]____ / df[denominator]____
return df