import random import math def gauss(balls, p): mu = p * balls sigma = math.sqrt((1-p) * p * balls) rand = round(random.gauss(mu, sigma)) if rand > balls: rand = balls if rand < 0: rand = 0 return rand ################### plist = [0.01, 0.02, 0.05, 0.03, 0.02, 0.08, 0.09, 0.11, 0.06, 0.01, 0.04, 0.33, 0.04, 0.02, 0.02, 0.05, 0.02] m = 60000 #################### anslist = [] sump = 1 balls = m for p in plist: pnow = p/(sump) answer = gauss(balls, pnow) print(answer) sump -= p balls -= answer