python - 这是获得随机舍入解决方案的正确方法吗？

Question

我想用Python实现随机舍入算法。

该算法很容易描述。

Given a vector, x, of size n where each element, x[i], of x is a real number between 0 and 1. Generate a 0-1 vector, y, of size n where each element of y, y[i], is either 0 or 1 as follows:

for i = 1 to len(x) do: set y[i] = 1 with probability x[i] and y[i] = 0 with probability 1 - x[i].

我尝试按如下方式执行此操作，但我想我错过了一些东西，或者也许有更快/更好的方法:

import numpy
n = 4
x = [0.01, 0.6, 0.1, 0.7]
y = numpy.zeros(n)
for i in range(n):
    randomNum = numpy.random.uniform()
    if randomNum >= x[i]:
        y[i] = 1
    else:
        y[i] = 0

Answer 1

这个怎么样:

x = np.array([0.1, 0.3,0.5, 0.7, 0.9])
y = np.random.rand(len(x)) <= x

第一行设置x，第二行随机均匀选择概率，并根据x[i]中的概率/阈值对结果进行二值化。