给定一个list的set及set中元素被选取的概率，从list中按set元素的概率抽取若干个数

Pywin

于 2020-11-25 15:01:12 发布

阅读量640

点赞数

分类专栏： python 文章标签：列表

本文链接：https://blog.csdn.net/qq_36076233/article/details/110130934

版权

python 专栏收录该内容

37 篇文章

订阅专栏

由于未提供博客具体内容，无法生成包含关键信息的摘要。

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

import random
import numpy as np
a = list(range(0, 9))   #a = [0,1,2,3,4,5,6,7,8]
a_ = list(range(0, 10)) #a_ = [0,1,2,3,4,5,6,7,8,9]
b = a * 5 + [9] * 2     #b = [0,1,2,3,4,5,6,7,8,...,9,9]
random.shuffle(b)

p1 = (1/5) / ((1/5)*9 + (1/2))
p2 = (1/2) / ((1/5)*9 + (1/2))
print(p1, p2)           #0.08695652173913045 0.2173913043478261

def random_choice(num_list, targets, size, p):
    targets_ = targets.copy()
    collector = []
    for i in range(size):
        sign = True
        ele = np.random.choice(num_list, 1, p)[0]
        while sign:
            if ele in targets_:
                collector.append(ele)
                targets_.remove(ele)
                sign = False
            else:
                ele = np.random.choice(num_list, 1, p)[0]
    return collector

samples = random_choice(a_, b, 10, p=[*([p1]*9), p2])
print(samples)         #[7, 1, 2, 9, 1, 5, 6, 4, 5, 9]