numpy.where() en Python

La función numpy.where() devuelve los índices de los elementos en una array de entrada donde se cumple la condición dada.

Sintaxis: numpy.where(condición[, x, y])
Parámetros:
condición: cuando es verdadero, genera x; de lo contrario, genera y.
x, y : Valores entre los que elegir. x, y y la condición deben poder transmitirse a alguna forma.

Devuelve:
out : [ndarray o tupla de ndarrays] Si se especifican tanto x como y, la array de salida contiene elementos de x donde la condición es verdadera y elementos de y en cualquier otro lugar.

Si solo se da la condición, devuelve la tupla condition.nonzero(), los índices donde condition es True.

Código #1:

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

Producción :

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

Código #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])

Producción :

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