Python numpy.ma.where()函数

Python numpy.ma.where()函数

numpy.ma.where()函数根据条件返回一个包含x或y元素的屏蔽数组。

语法: numpy.ma.where(condition, x, y)

参数:
condition : [array_like, bool] 其中True,产生x,否则产生y。
x, y : [array_like, optional] 可供选择的值。x, y和条件需要可广播到一些形状。

返回 : [MaskedArray] 一个带掩码元素的掩码数组,其中条件为掩码的元素来自x,条件为True的元素来自y,其他地方的元素来自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]]

Python教程

Java教程

Web教程

数据库教程

图形图像教程

大数据教程

开发工具教程

计算机教程

Numpy教程