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

鱼cpython学习者

举个例子，我有一个列表

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

复制代码

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

最佳答案

月排行榜 / 总排行榜

jackz007

2022-11-18 22:14:22

1. import random
   
2. 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

1. import random
   
2. 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 个元组被抽中的概率

复制代码

傻眼貓咪

1. import random
   
2. 
   
3. arr = [('a', 1), ('b', 3), ('c', 2), ('d', 3), ('e', 5)]
   
4. res = []
   
5. 
   
6. # 创建权重数组，数字越大，元素越多，被选中的几率越大
   
7. for each in arr:
   
8.     a, b = each
   
9.     for n in range(b):
   
10.         res.append(each)
    
11. 
    
12. # 将随机选中的元素存入数组 ans 里
    
13. ans = []
    
14. while len(ans) != 3:
    
15.     ans.append(random.choice(res))
    
16.     ans = list(set(ans)) # 确保元素不重复
    
17. 
    
18. print(ans) # 3 个无重复随机元素

复制代码