Générer vecteur un chaud à partir d'un tableau de NX1 d'étiquettes binaires

Question

J'ai un tableau comme celui-ci:

X = [0,0,1,1,0,0,1,1,1,0,0,0] 

Je veux créer un chaud vecteur codeur NX2

one_hotX = [[1,0],[1,0],[0,1],[0,1],[1,0]...] 

est-il un moyen facile de le faire? OneHotEncoder et LabelEncoder ne semblent pas fonctionner.

Answer 1

Que diriez-vous la construction du vecteur par vous-même:

[[0, 1] if i else [1, 0] for i in X] 

#[[1, 0], 
# [1, 0], 
# [0, 1], 
# [0, 1], 
# [1, 0], 
# [1, 0], 
# [0, 1], 
# [0, 1], 
# [0, 1], 
# [1, 0], 
# [1, 0], 
# [1, 0]] 

Si vous travaillez avec numpy, vous pourriez faire quelque chose comme cela aussi bien (approche vectorisé):

import numpy as np 
code = np.array([[1,0],[0,1]]) 
arrX = np.array(X) 

code[arrX] 

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