[已解决]有概率地随机选取列表里的元素

鱼cpython学习者

举个例子，我有一个列表

[('a', 1), ('b', 3), ('c', 2), ('d', 3), ('e', 5)]

现在要在里面随机选取三个元组，并且元组的第二个数字越大，选取到的概率越大
如何实现这样的算法？

最佳答案

月排行榜 / 总排行榜

jackz007

2022-11-18 22:14:22

import random
d = random . choices([('a', 1), ('b', 3), ('c', 2), ('d', 3), ('e', 5)] , [1 , 3 , 2 , 3 , 5] , k = 3)    # 参数 [1 , 3 , 2 , 3 , 5] 是权重，代表列表中 5 个元组被抽中的概率

跳转到最佳答案楼层

jackz007

import random
d = random . choices([('a', 1), ('b', 3), ('c', 2), ('d', 3), ('e', 5)] , [1 , 3 , 2 , 3 , 5] , k = 3)    # 参数 [1 , 3 , 2 , 3 , 5] 是权重，代表列表中 5 个元组被抽中的概率

傻眼貓咪

import random

arr = [('a', 1), ('b', 3), ('c', 2), ('d', 3), ('e', 5)]
res = []

# 创建权重数组，数字越大，元素越多，被选中的几率越大
for each in arr:
    a, b = each
    for n in range(b):
        res.append(each)

# 将随机选中的元素存入数组 ans 里
ans = []
while len(ans) != 3:
    ans.append(random.choice(res))
    ans = list(set(ans)) # 确保元素不重复

print(ans) # 3 个无重复随机元素