[英]Python mask for 2D array
I have generated a mask in the following manner- 我以以下方式生成了一个蒙版-
mask_v_co = numpy.ones((numRows_v_co, numCols_v_co)).astype(numpy.uint8)
counter = 0
for i in range(numRows_v_co):
for j in range(numCols_v_co):
if Data_v_co[i,j] < 0:
counter += 1 # Counting missing observation
mask_v_co[i,j] = 0
How can I generate a mask using numpy masked array module where 0 indicating invalid entries (wherever Data_v_co[i,j] < 0)
and 1 to indicate valid entries? 如何使用numpy掩码数组模块生成掩码,其中0表示无效条目(wherever Data_v_co[i,j] < 0)
和1表示有效条目?
Couldn't you just do something like the following? 您不可以做以下事情吗?
import numpy as np
mask = np.ones_like(Data_v_co, dtype='int8')
mask[Data_v_co < 0] = 0
# count zeros
counter = np.prod(mask.shape) - mask.sum()
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.