You could do this with randcraft
from randcraft.constructors import make_discrete
bernoulli = make_discrete(values=[0, 1], probabilities=[0.8, 0.2])
bernoulli_100 = bernoulli.multi_sample(100)
bernoulli_100.plot()
results = bernoulli_100.sample_numpy(5)
print(results)
# [10. 15. 20. 14. 24.]
