Python 按照概率分布生成随机数

导入库

import random

实用可直接调用函数

传入数组为概率分布列表例如[10, 90]，返回值为下标索引，返回值返回0的概率为10%，返回1的概率为90%

def random_index(rate):
    # """随机变量的概率函数"""
    # 参数rate为list<int>
    # 返回概率事件的下标索引
    start = 0
    index = 0
    randnum = random.randint(1, sum(rate))
    for index, scope in enumerate(rate):
        start += scope
        if randnum <= start:
            break
    return index

其他可以实用的随机函数以及讲解

random.randint(1,10)        # 产生 1 到 10 的一个整数型随机数  
random.random() )             # 产生 0 到 1 之间的随机浮点数
random.uniform(1.1,5.4)     # 产生  1.1 到 5.4 之间的随机浮点数，区间可以不是整数
random.choice('tomorrow')    # 从序列中随机选取一个元素
random.randrange(1,100,2)   # 生成从1到100的间隔为2的随机整数