python - Python 中来自值列表的加权随机数

Question

我正在尝试创建一个包含 1 到 1000 之间的 10,000 个随机数的列表。但我希望 80-85% 的数字属于同一类别(我的意思是其中大约 100 个数字应该出现 80% 的次数在随机数列表中)，其余的出现次数约为 15-20%。知道这是否可以在 Python/NumPy/SciPy 中完成。谢谢。

Answer 1

只需调用 1 次 random.randint() 即可轻松完成此操作选择一个列表并再次调用 random.choice()在正确的列表上。我假设列表 frequent 包含 100 个元素，这些元素将被选择 80 次，而 rare 包含 900 元素被选择 20 次。

import random
a = random.randint(1,5)
if a == 1:
    # Case for rare numbers
    choice = random.choice(rare)
else:
    # case for frequent numbers
    choice = random.choice(frequent)