Numpy 随机 0 和 1 矩阵，偏向 0

Question

是否有一种智能、快速的方法来创建一个充满 0 和 1 并偏向 0 的 n x n numpy 数组？

我做了 np.random.randint(2, (size,size)) 但这里没有考虑偏差。我可以做 for 循环，但我想要一种更快、更干净的方法来填充矩阵。谢谢！

Answer 1

我会使用

numpy.random.choice

与自定义概率：

n = 10
out = np.random.choice([0, 1], size=(n, n), p=[0.9, 0.1])

输出示例：

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