Numpy生成不超过N个的二进制数组

这是我能想到的最基本的方法:

import numpy as np

def binary_mask_random(r, c, n):
    a = np.zeros((r,c)).flatten()

    for i in range(np.random.randint(0, n+1)):
        x = np.random.randint(0, r*c)
        a[x] = 1

    return a.reshape((r,c))

1xr*c 零数组,并用最多 n 1 它在任意位置。返回一个 rxc