Python numpy.ma.where()用法及代码示例

numpy.ma.where()函数根据条件返回具有x或y元素的掩码数组。

用法： numpy.ma.where(condition, x, y)

参数：
condition:[数组，bool]如果为True，则产生x，否则产生y。
x, y:[数组，可选]从中选择的值。 x，y和条件必须广播到某种形状。

Return :[MaskedArray]一个带有被遮罩的元素(其中条件被遮罩)，来自x的元素(其中条件为True)以及来自y的元素的遮罩数组。

代码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)

输出：

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

代码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)

输出：

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

相关用法

注：本文由纯净天空筛选整理自sanjoy_62大神的英文原创作品 numpy.ma.where() function – Python。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。