004_021 Python 以指定的概率获取元素以一个列表为基准概率，从一个列表中随机获取元素

最新推荐文章于 2023-07-14 16:43:14 发布

书山登峰人

最新推荐文章于 2023-07-14 16:43:14 发布

阅读量1w

点赞数 1

CC 4.0 BY-SA版权

分类专栏： Python

本文链接：https://blog.youkuaiyun.com/houyj1986/article/details/22697213

Python 专栏收录该内容

113 篇文章

订阅专栏

本文介绍两种基于概率和权重实现的随机选择算法，并提供Python代码示例。一种算法使用均匀分布随机数并遍历列表，根据累积概率选取元素；另一种算法通过创建权重表，利用随机选择函数实现基于权重的选择。

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

代码如下：

#encoding=utf-8

print '中国'

#以指定的概率获取元素 以一个列表为基准概率，从一个列表中随机获取元素

import random

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

some_list = [1,2,3,4]
probabilities = [0.2,0.1,0.6,0.1]

print random_pick(some_list,probabilities)
    

#根据权重来获取 核心在于权重乘以 就相当于次数

def random_pick_odd(some_list, odds):
    print 'table2'
    table = [z for x,y in zip(some_list,odds) for z in [x] * y] 
    print table
    return random.choice(table)
    
some_list = [1,2,3,4]
odds=[3,1,4,2]

print random_pick_odd(some_list,odds)


#for item in random_picks()

打印结果如下：

中国
1
table2
[1, 1, 1, 2, 3, 3, 3, 3, 4, 4]
4