Начиная с версии 1.7.0, NumPy имеет функцию choice
, которая поддерживает распределения вероятностей.
from numpy.random import choice
draw = choice(list_of_candidates, number_of_items_to_pick, p=probability_distribution)
Обратите внимание, что probability_distribution
представляет собой последовательность в том же порядке из list_of_candidates
. Вы также можете использовать ключевое слово replace=False
, чтобы изменить поведение, чтобы нарисованные элементы не были заменены.