Самый быстрый способ заполнить массив numpy нулями и единицами

Question

Я хочу случайно заполнить массив numpy с произвольным размером с нулями и единицами. На каждой записи должен быть один с вероятностью 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

Вы хотите использовать Binomial distribution.

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