wie ein Wert numpy Array in bestimmten Zeilen und Spalten

Question

i ein Array zuweisen möchten zuweisen, dass bestimmte mit dem Wert (Zeile, Spalte) 1

hier mein Code:

fl = np.zeros((5, 3)) 
labels = np.random.random_integers(0, 2, (5, 1)) 
for i in range(5): 
    fl[i, labels[i]] = 1 

ist es eine Abkürzung für den Prozess?

Answer 1

Hier ist eine andere Art und Weise, es zu tun ist:

import numpy as np 
fl = np.zeros((5, 3)) 
labels = np.random.random_integers(0, 2, 5) 
fl[range(0, 5), labels] = 1 

Und es wird erzeugen Sie diesen Ausgang:

Answer 2

Sie können labels Array als boolesches Array mit fl.shape als Form verwenden. Versuchen:

import numpy as np 
fl = np.zeros((5, 3)) 
labels = np.random.random_integers(0, 1, fl.shape).astype(bool) 
fl[labels] = 1 

Und hier ist, wie die Anordnung von boolean der in Etiketten und Ergebnis wird wie folgt aussehen:

>>> labels 
array([[False, True, False], 
    [ True, True, False], 
    [False, True, True], 
    [ True, True, True], 
    [ True, False, False]], dtype=bool) 

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