主页 / coding / 问题 / 79002792
Accepted
Swaymon
Asked: 2024-09-19 21:30:31 +0800 CST

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

  • 772

是否有一种智能、快速的方法来创建一个填充有 0 和 1 且偏向于 0 的 nxn numpy 数组？

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

python

2 个回答

  1. Best Answer

    我将使用numpy.random.choice自定义概率：

    n = 10
p = 0.9 # probability of 0s

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

    另一种选择是，生成 0-1 范围内的数字，与 0 的概率进行比较并将布尔值转换为整数：

    out = (np.random.random((n, n))>p).astype(int)

    示例输出：

    [[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]]
    • 3

  2. 您可以使用np.random.rand然后检查值是否高于 0 到 1 之间的阈值。这将返回一个布尔数组，可以将其乘以1该数组以获得整数数组。

    1 * (np.random.rand(8, 8) > 0.8)

    返回：

    array([[0, 0, 1, 1, 1, 1, 0, 0],
       [1, 0, 0, 1, 0, 0, 1, 0],
       [0, 0, 0, 0, 0, 0, 0, 0],
       [0, 0, 0, 0, 0, 0, 1, 0],
       [0, 0, 1, 0, 1, 0, 0, 0],
       [1, 0, 0, 1, 0, 0, 0, 0],
       [0, 0, 0, 0, 0, 0, 0, 0],
       [0, 0, 0, 0, 0, 1, 0, 1]])
    • 2

相关问题