python - 확률로 빈에 정수를 분배하는 방법

어떻게 :

k = 10
l = [0.12, 0.12, 0.04, 0.04, 0.02, 0.02, 0.03, 0.03, 0.02, 0.02, 0.27, 0.27]
targets = [k*v for v in l]
new_l = [ int(v) for v in targets]
while sum(new_l) < k:
    residuals = [ t - v for t,v in zip(targets, new_l) ]
    index = residuals.index( max(residuals) )
    new_l[index] += 1

이것은 가장 큰 정수 n 를 찾는 것으로 시작합니다  그런 n   k*v 보다 작습니다 그런 다음 sum(new_l) == k 까지 최악의 n을 증가시켜 고정시킵니다. . 완료하는 데 O (len (l)) 이상의 작업이 필요하지 않습니다.

잘못 진행하고 있습니다. 당신이하려는 것은 당신의 목록에서 옵션을 선택하는 것입니다 n  타임스. 이것은 numpy.random.choice 를 사용하여 달성 할 수 있습니다  기능.

해당 코드는 다음과 같습니다.

from numpy.random import choice
from collections import Counter
draw = choice(range(12), 100,
              p=[0.12, 0.12, 0.04, 0.04, 0.02, 0.02, 0.03, 0.03, 0.02, 0.02, 0.27, 0.27])
counter = Counter(draw)
print(counter.values())
print(sum(counter.values()))

이것이 출력됩니다 :

dict_values([16, 13, 7, 3, 1, 1, 2, 2, 3, 2, 31, 19])
100

랜덤 화를 통해이를 수행 할 수 있습니다 :

sl = np.cumsum(l) # cumulative probability
b = np.random.rand(10) # a random number for each of your integers
new_l = np.zeros(l.shape)    
for i in range(k):
    iinsert = np.where(b[i]>sl)[0][0] # first entry where b is larger than the cumumlative prob
    new_l[iinsert] += 1