numpy.where() in Python

IL numpy.dove() la funzione restituisce gli indici degli elementi in un array di input in cui la condizione specificata è soddisfatta.

Sintassi: numpy.where(condizione[, x, y])
parametri:
condizione : Se Vero, ottieni x, altrimenti ottieni y.
x, y: Valori tra cui scegliere. x, y e la condizione devono essere trasmettibili in qualche forma.

Ritorna:
fuori : [ndarray o tupla di ndarray] Se vengono specificati sia x che y, l'array di output contiene elementi di x dove la condizione è True e elementi di y altrove.

Se viene fornita solo condizione, restituisce la tupla condizione.nonzero(), gli indici in cui condizione è Vero.

Codice n. 1:




# Python program explaining> # where() function> > import> numpy as np> > np.where([[> True> ,> False> ], [> True> ,> True> ]],> > [[> 1> ,> 2> ], [> 3> ,> 4> ]], [[> 5> ,> 6> ], [> 7> ,> 8> ]])>

Produzione :

array([[1, 6], [3, 4]]) 

Codice n. 2:




# Python program explaining> # where() function> > import> numpy as np> > # a is an array of integers.> a> => np.array([[> 1> ,> 2> ,> 3> ], [> 4> ,> 5> ,> 6> ]])> > print> (a)> > print> (> 'Indices of elements <4'> )> > b> => np.where(a <> 4> )> print> (b)> > print> (> 'Elements which are <4'> )> print> (a[b])>

Produzione :

[[1 2 3] [4 5 6]] Indices of elements  <4 (array([0, 0, 0], dtype=int64), array([0, 1, 2], dtype=int64)) Elements which are  <4 array([1, 2, 3])