from random import random
def bernoulli(p):
    if random()<p:
        return(1)
    else:
        return(0)

>>> bernoulli(0.4)
1
>>> bernoulli(0.8)
0