Schnellste Möglichkeit, numpy Array mit Nullen und Einsen zu füllen

Question

Ich möchte nach dem Zufallsprinzip ein numpy Array mit beliebiger Größe mit Nullen und Einsen füllen. Bei jedem Eintrag sollte eine Eins mit Wahrscheinlichkeit p sein.

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

Kennen Sie einen schnelleren Weg, dies zu tun?

Answer 1

Sie möchten die Binomial distribution verwenden.

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