Python以一定的概率生成某个数

比如以15%的概率生成数字0，以10%的概率生成数字1，以25%的概率生成数字2，以36%的概率生成数字3，以14%的概率生成数字4

代码如下：

import random


def number_of_certain_probability(sequence, probability):
    x = random.uniform(0, 1)
    cumulative_probability = 0.0
    for item, item_probability in zip(sequence, probability):
        cumulative_probability += item_probability
        if x < cumulative_probability:
            break
    return item

# 测试代码
value_list = [0, 1, 2, 3, 4]
probability = [0.15, 0.1, 0.25, 0.36, 0.14]
zero = 0
one = 0
two = 0
three = 0
four = 0
for i in range(1000):
    result = number_of_certain_probability(value_list, probability)
    if result == 1:
        one = one + 1
    elif result == 0:
        zero = zero + 1
    elif result == 2:
        two = two + 1
    elif result == 3:
        three = three + 1
    elif result == 4:
        four = four + 1
print(zero, one, two, three, four)

结果

进行1000次试验，结果符合预期。