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