pandas习题 195：把字符串类别一键编码

看过来

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

（编码题）把字符串类别一键编码 + 反向映射。8 月 6 条渠道订单：

import pandas as pd

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

用 一行 factorize() 完成：

把「渠道」列编码为 0, 1, 2
获得 反向映射（代码→标签）
打印编码结果和映射字典

参考答案

代码如下：

codes, uniques = pd.factorize(df["渠道"])
df["渠道编码"] = codes
mapping = dict(enumerate(uniques))

print("编码结果：\n", df)
print("映射字典：", mapping)

运行结果：

编码结果：
   订单号  渠道  渠道编码
0   D01  抖音       0
1   D02  快手       1
2   D03  微信       2
3   D04  抖音       0
4   D05  微信       2
5   D06  快手       1
映射字典： {0: '抖音', 1: '快手', 2: '微信'}

（完）

pandas习题 195：把字符串类别一键编码

参考答案

相关内容