Python中如何按独立概率随机选取任意数量的列表元素？

要实现每个元素按独立概率被选中的需求，核心思路是逐个检查每个元素，用随机数判断是否将其纳入结果，每个元素的判断过程互不影响。

基础实现代码 #

import random

example_list = ['Apple', 'Orange', 'Kiwi', 'Mango']
example_probabilities = [6, 9, 1, 4]  # 每个值对应元素被选中的百分比

selected_items = []
for item, prob_percent in zip(example_list, example_probabilities):
    # 把百分比转换为0-1的概率值，生成0-1的随机数进行判断
    if random.random() < prob_percent / 100:
        selected_items.append(item)

print(selected_items)

简洁写法（列表推导式） #

如果追求代码简洁，也可以用列表推导式实现相同逻辑：

import random

example_list = ['Apple', 'Orange', 'Kiwi', 'Mango']
example_probabilities = [6, 9, 1, 4]

selected_items = [
    item for item, prob_percent in zip(example_list, example_probabilities)
    if random.random() < prob_percent / 100
]

为什么不用random.choices()？ #

你提到的random.choices()确实不符合需求：它的权重是用来分配“选中某元素作为单次抽取结果”的概率，即每个元素的选中概率为权重/总权重，而且是指定抽取次数（k参数），结果里的元素是重复抽取或按比例选出的，并非每个元素独立判断是否被包含在结果中。

内容的提问来源于stack exchange，提问作者Om Goyal