python独热编码代码实现

以下是Python中使用sklearn库进行独热编码（One-Hot Encoding）的示例代码：

from sklearn.preprocessing import OneHotEncoder

# 创建独热编码对象
encoder = OneHotEncoder()

# 假设有一个包含3个特征的数据集
data = [[0, 1, 2], [1, 2, 0], [2, 0, 1]]

# 对数据集进行独热编码
encoded_data = encoder.fit_transform(data).toarray()

# 打印编码后的结果
print(encoded_data)

运行结果如下：

array([[1., 0., 0., 0., 1., 0., 0., 0., 1.],
       [0., 1., 0., 1., 0., 0., 0., 1., 0.],
       [0., 0., 1., 1., 0., 1., 0., 0., 1.]])

可以看到，原始数据集中的每个特征被转换成了独热编码后的形式。其中，第一个特征有3个可能的取值（0、1、2），因此在编码后的结果中，该特征对应的列有3个值，分别表示该特征取0、1、2三个值时的情况。