用零和一個填充numpy數組的最快方法

Question

我想隨機填充一個任意大小的numpy數組，其中包含0和1。 在每個條目上應該有一個概率為p的一個。

    for x in range(l):
        rn = numpy.random.uniform(0, 1)
        if rn <= p:
            arr[x] = 1.0
        else:
            arr[x] = 0.0

你知道更快的方法嗎？

Answer 1

您想要使用二項分布 。

arr = numpy.random.binomial(1, p, count)