función numpy.ma.where() – Python

numpy.ma.where()La función devuelve una array enmascarada con elementos de x o y, según la condición.

Sintaxis: numpy.ma.where (condición, x, y)

Parámetro:
condición: [array_like, bool] Donde es verdadero, produce x, de lo contrario, produce y.
x, y : [array_like, opcional] Valores entre los que elegir. x, y y la condición deben poder transmitirse a alguna forma.

Retorno: [MaskedArray] Una array enmascarada con elementos enmascarados donde la condición está enmascarada, elementos de x donde la condición es verdadera y elementos de y en cualquier otro lugar.

Código #1:

# Python program explaining
# numpy.ma.where() function
   
# importing numpy as geek 
# and numpy.ma module as ma 
import numpy as geek 
import numpy.ma as ma
    
x = geek.ma.array(geek.arange(4.).reshape(2, 2), 
                        mask =[[0, 1], [1, 0]])
  
gfg = geek.ma.where(x > 5, x, -3.1416)
   
print (gfg)

Producción :

[[-3.1416 --]
 [-- -3.1416]]

 
Código #2:

# Python program explaining
# numpy.ma.where() function
   
# importing numpy as geek 
# and numpy.ma module as ma 
import numpy as geek 
import numpy.ma as ma
    
x = geek.ma.array(geek.arange(9.).reshape(3, 3), 
       mask =[[0, 1, 0], [1, 0, 1], [0, 1, 0]])
  
gfg = geek.ma.where(x > 5, x, -3.1416)
   
print (gfg)

Producción :

[[-3.1416 -- -3.1416]
 [-- -3.1416 --]
 [6.0 -- 8.0]]