以概率列表选择对应元素,轮盘概率选择Python实现

做法和其他人一样,搞了半天明白了,ran = random.random(),是产生0———1之间的随机浮点数,
即 如下代码,第一个随机生成的数小于0.1的概率就是 10%,若第一个数大于0.1剩下的就是概率中先拿和后拿问题
import random
def select():
    num_ = ['a', 'b', 'c']
    #概率列表
    r_ = [0.1, 0.3, 0.6]
    sum_ = 0
    ran = random.random()
    for num, r in zip(num_, r_):
        sum_ += r
        if ran < sum_ :break
    return num
"""
验证结果,多次循环把a,b,c单独出现的次数计数,判断对应概率
"""
num_a, num_b ,num_c = 0, 0, 0
for each in range(100000):
    result = select()
    if result == 'a':
        num_a += 1
    elif result == 'b':
        num_b += 1
    else:
        num_c += 1
        
print(num_a, num_b, num_c)

结果如下还是符合预期的:

9983 29973 60044

 

posted @ 2017-10-10 10:30  曙暮之光  阅读(3464)  评论(0编辑  收藏  举报