Оновлення користувацької функції для виконання віднімання

Функція, над якою ви працювали у попередній вправі, не вміє віднімати. Розгляньте цю функцію:

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

Ця функція може виконувати додавання та віднімання в чисельниках і знаменниках фінансових коефіцієнтів. Зверніть увагу, що у функції використано np.where. Це функція з пакета NumPy. np.where перевіряє, чи перший аргумент є True; якщо так, повертає другий аргумент, інакше повертає третій. Наприклад, вище ми маємо:

np.where(addition_in_numerator,
                             df[numerator].sum(axis=1), 
                             df[numerator[0]] - df[numerator[1:]].sum(
                               axis=1))

Якщо addition_in_numerator дорівнює True, np.where поверне df[numerator].sum(axis=1), інакше поверне df[numerator[0]] - df[numerator[1:]].sum(axis=1).

У цій вправі датафрейм balance_sheet, а також pandas і NumPy як pd і np відповідно вже завантажені для вас. Скористайтеся ними, щоб визначити, яке з наведених тверджень є правильним.