python类别变量怎么转化为数值型

92次阅读
没有评论

共计 630 个字符,预计需要花费 2 分钟才能阅读完成。

在 Python 中,可以使用独热编码(One-Hot Encoding)将类别变量转化为数值型。

独热编码是一种将类别变量转化为二进制向量的方法。对于一个有 n 个类别的变量,独热编码将其转化为 n 个二进制变量,其中每个变量代表一个类别,如果数据点属于该类别,则对应的二进制变量为 1,否则为 0。

可以使用 scikit-learn 库中的 OneHotEncoder 来进行独热编码,示例如下:

from sklearn.preprocessing import OneHotEncoder

# 定义类别变量
categories = ['A', 'B', 'C', 'A', 'B', 'C']

# 创建 OneHotEncoder 对象
encoder = OneHotEncoder(sparse=False)

# 将类别变量进行独热编码
encoded_categories = encoder.fit_transform(np.array(categories).reshape(-1, 1))

print(encoded_categories)

输出结果为:

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

每一行表示一个数据点的独热编码,其中 1 的位置代表该数据点所属的类别。

丸趣 TV 网 – 提供最优质的资源集合!

正文完
 
丸趣
版权声明:本站原创文章,由 丸趣 2023-12-16发表,共计630字。
转载说明:除特殊说明外本站除技术相关以外文章皆由网络搜集发布,转载请注明出处。
评论(没有评论)