使用自訂函式計算多個比率

來看看你在前兩個練習中看到的這個函式。

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

回想上一個練習，我們用這個函式來計算比率。然而，就計算單一比率而言，使用這個函式並沒有更有效率，也沒有更少的程式碼。在本練習中，你會看到如何把這個函式用在迴圈中，一次計算多個比率。這樣做在計算多個比率時會更有效率，且需要的程式碼更少。