python - 如何在python中生成一个小于或等于另一个矩阵的随机矩阵

Question

我想在python中生成一个小于或等于另一个矩阵的随机0-1矩阵。例如，我有一个特定的矩阵：

a=[1 0 0
   0 1 1
   0 0 1]
                                   

我想生成一个随机矩阵：

y=[1 0 0
   0 0 0 
   0 0 1]

正如我们所见，a>=y。如何在 python 中生成随机矩阵 y？

标签： pythonpython-3.x

Answer 1

我会说生成一个随机矩阵，然后设置为 0 值太高

import numpy as np

a = np.array([[1, 0, 0], [0, 1, 1], [0, 0, 1]])
print(a)

y = np.random.randint(2, size=a.shape)
y[y > a] = 0
print(y)