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:
输出 :
代码#2:
输出 :