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

liulicuican

于 2019-10-11 15:46:28 发布

阅读量1.1w

点赞数 14

文章标签： python 以指定概率生成数字

本文链接：https://blog.csdn.net/liulicuican/article/details/102502540

版权

本文介绍了一种基于概率的数字生成算法，通过随机数与累积概率的比较，实现不同数字按预设概率生成。代码示例展示了如何使用Python实现这一算法，并通过1000次试验验证了结果的准确性。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

比如以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)