Costruire matrice bidimensionale NumPy da indici ei valori di un array monodimensionale

Question

Say Ho

Y = np.array([2, 0, 1, 1]) 

Da questo voglio ottenere una matrice X di forma (len(Y), 3). In questo caso particolare, la prima riga di X dovrebbe avere uno sul secondo indice e zero altrowhise. La seconda riga di X dovrebbe avere uno sull'indice 0 e zero altrimenti. Per essere espliciti:

X = np.array([[0, 0, 1], [1, 0, 0], [0, 1, 0], [0, 1, 0]]) 

Come si genera questa matrice? Ho iniziato con

X = np.zeros((Y.shape[0], 3)) 

ma poi non riusciva a capire come popolare/riempire quelli dalla lista degli indici

Come sempre, grazie per il vostro tempo!

Answer 1

Forse:

>>> Y = np.array([2, 0, 1, 1]) 
>>> X = np.zeros((len(Y), 3)) 
>>> X[np.arange(len(Y)), Y] = 1 
>>> X 
array([[ 0., 0., 1.], 
     [ 1., 0., 0.], 
     [ 0., 1., 0.], 
     [ 0., 1., 0.]]) 

Answer 2

Y = np.array([2, 0, 1, 1]) 
new_array = np.zeros((len(Y),3)) 
for i in range(len(Y)): 
    new_array[i,Y[i]] = 1 

credo ... io non credo che ci sia un modo più semplice (ma potrei sbagliarmi)

Answer 3

Per dare un one-liner alternativa al DSM di perfezione buona risposta:

>>> Y = np.array([2, 0, 1, 1]) 
>>> np.arange(3) == Y[:, np.newaxis] 
array([[False, False, True], 
     [ True, False, False], 
     [False, True, False], 
     [False, True, False]], dtype=bool)