Pandas 独热编码：使用 pd.get_dummies() 将分类变量转换为数值

要使用 pd.get_dummies()，你需要导入 pandas 库并创建一个 DataFrame 对象，然后将该对象作为参数传递给 pd.get_dummies() 方法。

下面是一个示例代码，演示如何使用 pd.get_dummies()：

import pandas as pd

# 创建一个包含分类变量的DataFrame
data = {'color': ['红', '蓝', '绿', '红', '红', '蓝']}
df = pd.DataFrame(data)

# 使用pd.get_dummies()对分类变量进行独热编码
dummies = pd.get_dummies(df['color'])

# 将独热编码结果与原始DataFrame进行合并
df = pd.concat([df, dummies], axis=1)

print(df)

运行上述代码会输出以下结果：

  color  红  绿  蓝
0    红  1  0  0
1    蓝  0  0  1
2    绿  0  1  0
3    红  1  0  0
4    红  1  0  0
5    蓝  0  0  1

在上述示例中，我们使用 pd.get_dummies() 对 color 列进行独热编码，并将编码结果与原始的 DataFrame 进行合并。最终的结果包含了原始 color 列以及独热编码后的三个新列。