pandas习题 188：把哑变量「渠道」还原分类变量

看过来

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

（编码题）把哑变量还原为原始分类列。运营部拿到一张 已经哑变量化 的 DataFrame：

import pandas as pd

dummies = pd.DataFrame({
    "订单号": ["D01", "D02", "D03", "D04"],
    "渠道_抖音": [1, 0, 0, 1],
    "渠道_快手": [0, 1, 0, 0],
    "渠道_微信": [0, 0, 1, 0]
})

用 一行 pd.from_dummies() 把哑变量列还原成 原始“渠道”列（前缀为 "渠道"），并打印结果。

参考答案

代码如下：

restored = pd.from_dummies(dummies, prefix="渠道")
print(restored)

输出：

  订单号 渠道
0  D01  抖音
1  D02  快手
2  D03  微信
3  D04  抖音

（完）

pandas习题 188：把哑变量「渠道」还原分类变量

参考答案

相关内容