pandas习题 144：apply() 调用外部函数

看过来

《pandas 教程》持续更新中，提供建议、纠错、催更等加作者微信: gr99123（备注：pandas教程）和关注公众号「盖若」ID: gairuo。跟作者学习，请进入 Python学习课程。欢迎关注作者出版的书籍：《深入浅出Pandas》和《Python之光》。

（编码题）有这样一个商品表：

import pandas as pd

df = pd.DataFrame({
    "商品": ["A", "B", "C"],
    "单价": [10, 20, 30],
    "数量": [2, 3, 1]
})

写一个外部函数 calc_total(row)，计算 单价 * 数量，然后用 一行 apply() 新增列 总价。

代码

代码如下：

def calc_total(row):
    return row['单价'] * row['数量']

df['总价'] = df.apply(calc_total, axis=1)
print(df)

输出：

  商品  单价  数量  总价
0    A   10     2   20
1    B   20     3   60
2    C   30     1   30

（完）