gpt4 book ai didi

python - 加权选择简短而简单

转载 作者:IT老高 更新时间:2023-10-28 20:35:52 25 4
gpt4 key购买 nike

如果我有一个列表中的项目集合。我想根据另一个权重列表从该列表中进行选择。

例如我的集合是 ['one', 'two', 'three'] 权重是 [0.2, 0.3, 0.5],我会希望该方法在大约一半的抽奖中给我“三”。

最简单的方法是什么?

最佳答案

自从 1.7版你可以使用numpy.random.choice() :

elements = ['one', 'two', 'three'] 
weights = [0.2, 0.3, 0.5]

from numpy.random import choice
print(choice(elements, p=weights))

关于python - 加权选择简短而简单,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/10803135/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com