栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 面试经验 > 面试问答

用Python方式选择具有不同概率的列表元素

面试问答 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

用Python方式选择具有不同概率的列表元素

权重定义概率分布函数(pdf)。任何此类pdf的随机数均可通过将其关联的逆累积分布函数应用于0到1之间的均匀随机数来生成。

另请参见以下SO解释,或者如Wikipedia所述:

如果Y具有U [0,1]分布,则F⁻(Y)作为F分布。这用于使用逆变换采样方法的随机数生成中。

import randomimport bisectimport collectionsdef cdf(weights):    total = sum(weights)    result = []    cumsum = 0    for w in weights:        cumsum += w        result.append(cumsum / total)    return resultdef choice(population, weights):    assert len(population) == len(weights)    cdf_vals = cdf(weights)    x = random.random()    idx = bisect.bisect(cdf_vals, x)    return population[idx]weights=[0.3, 0.4, 0.3]population = 'ABC'counts = collections.defaultdict(int)for i in range(10000):    counts[choice(population, weights)] += 1print(counts)# % test.py# defaultdict(<type 'int'>, {'A': 3066, 'C': 2964, 'B': 3970})

choice
上述用途功能
bisect.bisect
,所以加权随机变量的选择是在完成
O(log n)
其中
n
是的长度
weights


请注意,从1.7.0版开始,NumPy具有Cythonized
np.random.choice函数。例如,这从

[0,1,2,3]
具有权重的总体中生成1000个样本
[0.1,0.2, 0.3, 0.4]

import numpy as npnp.random.choice(4, 1000, p=[0.1, 0.2, 0.3, 0.4])

np.random.choice
也有一个
replace
参数,可以选择是否进行替换。


理论上更好的算法是Alias方法。它会建立一个需要

O(n)
时间的表格,但此后可以及时绘制样本
O(1)
。因此,如果您需要绘制许多样本,则从理论上讲,别名方法可能会更快。有一个Python实现沃克别名方法在这里和这里numpy的版本。



转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/638805.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号