อัปเดตฟังก์ชันที่กำหนดเองให้รองรับการลบ

ฟังก์ชันที่ใช้ในแบบฝึกหัดก่อนหน้ายังไม่รองรับการลบ ลองดูฟังก์ชันนี้:

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 เป็นจริง np.where จะคืนค่า df[numerator].sum(axis=1) มิฉะนั้นจะคืนค่า df[numerator[0]] - df[numerator[1:]].sum(axis=1)

ในแบบฝึกหัดนี้ DataFrame balance_sheet พร้อมด้วย pandas และ NumPy ในชื่อ pd และ np ตามลำดับ ได้ถูกโหลดไว้ให้แล้ว ใช้ข้อมูลเหล่านี้เพื่อพิจารณาว่าข้อความใดต่อไปนี้ถูกต้อง