Construct dwuwymiarowej tablicy numpy z indeksów i wartości w jednej tablicy wymiarowej

Question

Say mam

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

Od tego chcę otrzymać macierz X w kształcie (len(Y), 3). W tym konkretnym przypadku pierwszy wiersz X powinien mieć jeden na drugim indeksie, a zero na inne. Drugi wiersz X powinien mieć jeden na indeksie 0, a zero w przeciwnym razie. Wyraźnie:

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

Jak mogę wyprodukować tę matrycę? Zacząłem

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

ale potem nie mógł dowiedzieć się, jak wypełnić/wypełnić te z listy indeksów

jak zawsze, dzięki za poświęcony czas!

Answer 1

Może:

>>> 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 

myślę ... nie sądzę że istnieje prostszy sposób (ale może się mylę)

Answer 3

Aby dać alternatywę jedno-liner do DSM doskonale dobra odpowiedź:

>>> 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)