Question

我尝试了不同的方法，但似乎无法无循环地高效地进行操作。

输入是一个数组y和一个百分比x。

例如输入为

y=np.random.binomial(1,1,[10,10])
x=0.5

输出

[[0 0 0 0 1 1 1 1 0 1]
 [1 0 1 0 0 1 0 1 0 1]
 [1 0 1 1 1 1 0 0 0 1]
 [0 1 0 1 1 0 1 0 1 1]
 [0 1 1 0 0 1 1 1 0 0]
 [0 0 1 1 1 0 1 1 0 1]
 [0 1 0 0 0 0 1 0 1 1]
 [0 0 0 1 1 1 1 1 0 0]
 [0 1 1 1 1 0 0 1 0 0]
 [1 0 1 0 1 0 0 0 0 0]]

Answer 1

根据您要执行的操作，有几种矢量化方法可能会对您有所帮助：

# Flatten the 2D array and get the indices of the non-zero elements
c = y.flatten()
d = c.nonzero()[0]
# Shuffle the indices and set the first 100x % to zero
np.random.shuffle(d)
x = 0.5
c[d[:int(x*len(d))]] = 0
# reshape to the original 2D shape
y = c.reshape(y.shape)

毫无疑问，这里需要进行一些效率改进。

Answer 2

这是基于masking-

def set_nonzeros_to_zeros(a, setz_ratio):
    nz_mask = a!=0
    nz_count = nz_mask.sum()
    z_set_count = int(np.round(setz_ratio*nz_count))

    idx = np.random.choice(nz_count,z_set_count,replace=False)
    mask0 = np.ones(nz_count,dtype=bool)
    mask0.flat[idx] = 0
    nz_mask[nz_mask] = mask0
    a[~nz_mask] = 0
    return a

我们跳过了np.argwhere/np.nonzero的所有索引的生成，而转而使用基于掩码的索引来关注性能。

样品运行-

In [154]: np.random.seed(0)
     ...: a = np.random.randint(0,3,(5000,5000))

# number of non-0s before using solution
In [155]: (a!=0).sum()
Out[155]: 16670017

In [156]: a_out = set_nonzeros_to_zeros(a, setz_ratio=0.2) #set 20% of non-0s to 0s

# number of non-0s after using solution
In [157]: (a_out!=0).sum()
Out[157]: 13336014

# Verify
In [158]: 16670017 - 0.2*16670017
Out[158]: 13336013.6

在numpy 2d数组中设置为非零值的0 x％

2 个答案: