Calculando múltiplos índices com a função definida pelo usuário

Vamos dar uma olhada na função que você viu nos dois últimos exercícios.

def compute_ratio(df, numerator, denominator, ratio_name, 
                  addition_in_numerator = True,
                  addition_in_denominator = True):
  ratio_numerator = np.where(addition_in_numerator,
                             df[numerator].sum(axis=1), 
                             df[numerator[0]] - df[numerator[1:]].sum(
                               axis=1))
  ratio_denominator = np.where(addition_in_denominator, 
                               df[denominator].sum(axis=1), 
                               df[denominator[0]] - df[denominator[1:]].sum(axis=1))
  df[ratio_name] = ratio_numerator/ratio_denominator
  return df

Lembre que, no exercício anterior, usamos a função para calcular índices. Ainda assim, não foi mais eficiente nem envolveu menos código do que calcular os índices sem a função. Neste exercício, você vai ver como usar a função para calcular vários índices em um loop. Isso torna o cálculo de múltiplos índices mais eficiente e exige menos código.