如何用Python实现基于概率的抽球函数（无需重复元素数组）

不用重复元素的数组完全可以实现需求，核心思路是按颜色对应的概率分配随机数区间，不用把所有球都放进数组里，既节省空间逻辑也清晰，尤其适合基础阶段理解概率模拟的本质。

具体实现逻辑：

1. 先统计每种颜色球的数量和总球数：总共有5蓝+2黄+1粉=8个球
2. 给每种颜色分配对应数量的随机数范围：
   • 蓝色对应0~4（共5个数字，和球数一致）
   • 黄色对应5~6（共2个数字）
   • 粉色对应7（1个数字）
3. 生成0到7之间的随机整数，判断它落在哪个区间，就对应取出哪种颜色的球

举个Python的代码例子（用你提到的["blue", "yellow", "pink"]数组实现）：

import random

def pick_ball():
    colors = ["blue", "yellow", "pink"]
    color_counts = [5, 2, 1]
    total_balls = sum(color_counts)
    
    # 生成0到总球数-1的随机数
    rand_index = random.randint(0, total_balls - 1)
    
    # 遍历判断随机数对应的颜色
    cumulative_count = 0
    for color, count in zip(colors, color_counts):
        cumulative_count += count
        if rand_index < cumulative_count:
            return color
    # 兜底返回最后一种颜色（理论上不会走到这里）
    return colors[-1]

# 测试取球
print(pick_ball())

这种方式比用重复元素的数组（比如["blue"]*5 + ["yellow"]*2 + ["pink"]）更高效，尤其是当球的数量极大时，不会占用多余内存；同时维护性更强——要修改某颜色的球数，直接改数字即可，不用调整数组长度。

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