在NumPy中将索引数组转换为独热编码数组

Question

在NumPy中将索引数组转换为独热编码数组

pythonnumpymachine-learningnumpy-ndarrayone-hot-encoding

354

给定一个一维索引数组：

a = array([1, 0, 3])

我想将其作为一个二维数组进行one-hot编码：

b = array([[0,1,0,0], [1,0,0,0], [0,0,0,1]])

- James Atwood

22个回答

网页内容由stack overflow 提供, 点击上面的

可以查看英文原文，
原文链接

- Inaam Ilahi · Answer 1

使用以下代码。它最好运行。

def one_hot_encode(x):
"""
    argument
        - x: a list of labels
    return
        - one hot encoding matrix (number of labels, number of class)
"""
encoded = np.zeros((len(x), 10))

for idx, val in enumerate(x):
    encoded[idx][val] = 1

return encoded

在这里找到它 P.S 你不需要进入链接。

- Guillaume Chevalier · Answer 2

使用 Neuraxle 管道步骤：

设置您的示例

import numpy as np
a = np.array([1,0,3])
b = np.array([[0,1,0,0], [1,0,0,0], [0,0,0,1]])

进行实际转换

from neuraxle.steps.numpy import OneHotEncoder
encoder = OneHotEncoder(nb_columns=4)
b_pred = encoder.transform(a)

确保它能正常工作

assert b_pred == b

文档链接: neuraxle.steps.numpy.OneHotEncoder