Python - prawdopodobieństwo modelowania

Question

Mam prosty problem. Potrzebuję sposobu, aby utworzyć funkcję, która generuje 0 w p procentach przypadków i 1 we wszystkich innych przypadkach. Próbowałem to robić z random.Random() jak poniżej:

p = 0.40 

def generate(): 
    x = random.random() 
    if x < p: 
     return 0 
    else: 
     return 1 

to jednak nie wydaje się dobrym sposobem. A może jest?

Answer 1

Twoja obecna metoda jest całkowicie w porządku, możesz to sprawdzić, wykonując kilka prób z wieloma próbami. Na przykład oczekiwalibyśmy około 600000 Prawdziwych wyników przy 1000000 próbach:

>>> sum(generate() for i in range(1000000)) 
599042 
>>> sum(generate() for i in range(1000000)) 
599670 
>>> sum(generate() for i in range(1000000)) 
600011 
>>> sum(generate() for i in range(1000000)) 
599960 
>>> sum(generate() for i in range(1000000)) 
600544 

Wygląda poprawnie.

Jak zauważył w komentarzach, można skrócić metoda trochę:

p = 0.40 

def generate(): 
    return random.random() >= p 

Jeśli chcesz 1 i 0 zamiast True i False można użyć int(random.random() >= p), ale jest to prawie na pewno konieczne.