如何基于Python字典中的百分比权重生成符合概率的随机水果列表

原代码存在的问题 #

• 缩进错误：Python 对函数体内的代码缩进有严格要求，你写的 import random、循环逻辑、return 语句都没有正确缩进，会直接触发语法错误
• 返回值类型不符需求：你初始化 fruit_list 为空字符串，但你的需求是生成列表，应该初始化为空列表 []，追加元素用 append() 方法而非字符串拼接
• 概率区间设置错误：所有水果的概率总和是 100%，但你用 random.uniform(0,99) 生成随机数，会导致香蕉的实际概率被压缩到约 14%，应该改为 random.uniform(0, 100)
• 硬编码逻辑脆弱：所有判断条件都写死了字典键名，后续如果字典增减水果类型，需要大幅修改逻辑，维护成本高
• 内容错误：返回的水果名存在拼写、大小写不一致的问题，比如 Bannana 应为 Banana，Grapes 和其他小写名称风格不统一

修复后的基础版本（贴合你原本的编写思路） #

import random

def generate_list(length, fruits):
    fruit_list = []
    # 提前计算区间边界，避免循环内重复计算
    orange_bound = fruits['Oranges%']
    apple_bound = orange_bound + fruits['Apples%']
    grape_bound = apple_bound + fruits['Grapes%']
    for i in range(length):
        num = random.uniform(0, 100)
        if num < orange_bound:
            fruit_list.append('Oranges')
        elif num < apple_bound:
            fruit_list.append('Apples')
        elif num < grape_bound:
            fruit_list.append('Grapes')
        else:
            fruit_list.append('Bananas')
    return fruit_list

更通用的优化版本（适配任意水果字典） #

无需写死键名，只要字典符合「名称%: 概率值」的格式就能自动适配，代码更简洁：

import random

def generate_list(length, fruits):
    fruit_list = []
    # 自动提取水果名和对应概率权重
    fruit_names = [k.replace('%', '') for k in fruits.keys()]
    weights = list(fruits.values())
    for _ in range(length):
        # 直接用内置方法按权重随机抽取，无需手动写区间判断
        picked = random.choices(fruit_names, weights=weights, k=1)[0]
        fruit_list.append(picked)
    return fruit_list

注：random.choices 是 Python 3.6 及以上版本支持的内置方法，传入 weights 参数即可直接按给定权重随机抽取元素，实现效率更高也不容易出错。

测试示例 #

fruits = { 'Oranges%' : 30.0, 
           'Apples%' :  20.0,
           'Grapes%' : 35.0,
           'Bananas%' : 15.0}

# 生成长度为10的水果列表
print(generate_list(10, fruits))

运行后会输出类似 ['Grapes', 'Oranges', 'Bananas', 'Apples', 'Grapes', 'Grapes', 'Oranges', 'Apples', 'Grapes', 'Oranges'] 的结果，符合设定的概率分布要求。

内容的提问来源于stack exchange，提问作者Pandora core