使用用户自定义函数计算多个比率

来看一下您在前两个练习中见过的这个函数。

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

回顾上一个练习，我们用这个函数来计算比率，但当时并没有更高效，也没有减少多少代码量。本练习将向您展示如何在循环中使用该函数一次性计算多个比率。这样一来，计算多个比率会更高效，所需代码也更少。