Fastest way to fill numpy array with zeros and ones

Question

I want to fill randomly a numpy array with arbitrary size with zeroes and ones. On each entry there should be a one with probability p.

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

Do you know a faster way to do this?

kennytm · Accepted Answer · 2016-05-12 12:14:32Z

8

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

answered May 12, 2016 at 12:14

kennytm

526k110 gold badges1.1k silver badges1k bronze badges

Sign up to request clarification or add additional context in comments.

1 Answer 1