更新用户自定义函数
用户自定义函数可以大幅减少重复工作。视频中定义的函数如下:
def compute_ratio(df, numerator, denominator,
ratio_name):
df[ratio_name] = df[numerator] / df[denominator]
return df
请注意,上面的函数无法直接计算需要在分子或分母中先进行加总或相减的比率,例如营业利润率或毛利率。
在本练习中,您将更新该函数,使其能够计算在分子和分母中涉及「加法」的比率。
您将使用 pandas 的 .sum() 函数,它可以沿着 DataFrame 的某个轴求和。轴为 0 表示按行方向求和,因此会对每一列沿行累加并返回每列的一个值。轴为 1 表示按列方向求和,因此会对 df 中不同列的值相加,并为每一行返回一个值。您可以把 df.sum(axis=1) 理解为等价于 df[column_1] + ... + df[column_n]。在 pd.sum() 中,默认的 axis 为 0。
本练习是课程的一部分
用 Python 分析财务报表
交互式实操练习
通过完成这段示例代码来试试这个练习。
# Update the function
def compute_ratio(df, numerator, denominator, ratio_name):
df[ratio_name] = df[numerator]____ / df[denominator]____
return df