Я хочу провести относительно простую случайную ничью в numpy, но не могу найти хороший способ выразить это. Я думаю, что лучший способ это описать как рисунок из урны без замены. У меня есть урна с k цветами и n_k шарами каждого цвета. Я хочу нарисовать m шаров, и знаю, сколько шаров каждого цвета у меня есть.

Моя текущая попытка это

np.bincount(np.random.permutation(np.repeat(np.arange(k), n_k))[:m], minlength=k)

Вот,n_k является массивом длины k с количеством шаров.

Кажется, это эквивалентноnp.bincount(np.random.choice(k, m, n_k / n_k.sum(), minlength=k)

что немного лучше, но все же не отлично.