pandas习题 177：「城市→大区」映射一键聚合

看过来

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

（编码题）用“城市→大区”映射一键聚合。运营部拿到 8 月 6 个城市的 GMV：

import pandas as pd

df = pd.DataFrame({
    "城市": ["北京", "上海", "广州", "深圳", "成都", "杭州"],
    "GMV": [5000, 4800, 3200, 3000, 2800, 2600]
})

# 城市→大区映射
map_dict = {
    "北京": "华北",
    "上海": "华东",
    "广州": "华南",
    "深圳": "华南",
    "成都": "西南",
    "杭州": "华东"
}

用 一行链式代码 完成：

按 字典映射 把城市归并为大区
计算每大区：
- GMV 总和
- 城市个数
结果列名：总GMV、城市数

参考答案

代码如下：

(
    df.set_index('城市')
    .groupby(map_dict)
    .GMV
    .agg(总GMV="sum", 城市数="count")
)

输出示例：

     总GMV  城市数
大区           
华北   5000      1
华东   7400      2
华南   6200      2
西南   2800      1

（完）

pandas习题 177：「城市→大区」映射一键聚合

参考答案

相关内容