pandas习题 187：把「渠道」转为哑变量

看过来

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

（编码题）运营部 6 条订单：

import pandas as pd

df = pd.DataFrame({
    "订单号": ["D01", "D02", "D03", "D04", "D05", "D06"],
    "渠道":   ["抖音", "快手", "微信", "抖音", "微信", "快手"],
    "GMV":    [1000, 800, 600, 1200, 700, 900]
})

用 一行 pd.get_dummies() 把“渠道”转为哑变量（前缀为“渠道”），并直接合并回原表。

参考答案

代码如下：

df = pd.get_dummies(df, columns=["渠道"], prefix="渠道")
print(df)

输出：

  订单号  GMV  渠道_抖音  渠道_快手  渠道_微信
0  D01  1000        1        0        0
1  D02   800        0        1        0
2  D03   600        0        0        1
3  D04  1200        1        0        0
4  D05   700        0        0        1
5  D06   900        0        1        0

（完）

pandas习题 187：把「渠道」转为哑变量

参考答案

相关内容