使用Python从CSV文件抽取随机中奖者：基于票数权重实现

Name,Ticket count
Alice,3
Bob,2
Charlie,4
Dana,1

import csv
import random

def generate_winner_bag(filepath: str) -> list[str]:
    """
    从CSV文件读取参与者及其票数，并构建一个“名字袋”列表。

    Args:
        filepath (str): CSV文件的路径。

    Returns:
        list[str]: 包含所有加权姓名的列表（名字袋）。
    """
    bag_of_names: list[str] = []

    try:
        with open(filepath, newline="", encoding="utf-8") as f:
            reader = csv.reader(f, skipinitialspace=True)
            # 跳过CSV文件的标题行（如果存在）
            header = next(reader, None) 
            if header is None:
                print(f"警告：文件 '{filepath}' 似乎为空或不含数据。")
                return []

            for row in reader:
                if len(row) < 2:
                    print(f"警告：跳过格式不正确的行：{row}")
                    continue

                name = row[0].strip() # 清除姓名两边的空白
                try:
                    ticket_ct = int(row[1].strip()) # 清除票数两边的空白并转换为整数
                    if ticket_ct <= 0:
                        print(f"警告：参与者 '{name}' 的票数 '{ticket_ct}' 无效，已忽略。")
                        continue
                    # 使用 extend 和列表乘法高效地添加名字
                    bag_of_names.extend([name] * ticket_ct)
                except ValueError:
                    print(f"警告：参与者 '{name}' 的票数 '{row[1]}' 不是有效数字，已忽略。")
                    continue
    except FileNotFoundError:
        print(f"错误：文件 '{filepath}' 未找到。请检查路径。")
    except Exception as e:
        print(f"读取文件时发生意外错误：{e}")

    return bag_of_names

def pick_winner(bag_of_names: list[str]) -> str:
    """
    从给定的“名字袋”中随机抽取一名中奖者。

    Args:
        bag_of_names (list[str]): 包含所有加权姓名的列表。

    Returns:
        str: 随机选中的中奖者姓名。
    """
    if not bag_of_names:
        return "没有可供抽取的参与者。"
    return random.choice(bag_of_names)

if __name__ == "__main__":
    csv_file_path = "raffle.csv" # 确保此路径指向您的CSV文件

    # 1. 构建名字袋
    winner_bag = generate_winner_bag(csv_file_path)

    if winner_bag:
        # 2. 抽取中奖者
        winner = pick_winner(winner_bag)
        print(f"恭喜！本次抽奖的中奖者是：{winner}")

        # 示例：多次抽奖以验证公平性 (可选)
        from collections import Counter
        num_simulations = 1000 # 模拟抽奖次数

        if len(set(winner_bag)) > 1: # 只有当有多个不同名字时才进行模拟
            print(f"
进行 {num_simulations} 次模拟抽奖以验证公平性：")
            simulation_results: Counter[str] = Counter()
            for _ in range(num_simulations):
                simulation_results[pick_winner(winner_bag)] += 1

            total_entries = len(winner_bag)
            print("--- 模拟结果 ---")
            for name, wins in simulation_results.most_common():
                # 计算每个参与者的理论中奖概率（基于其票数）
                theoretical_probability = winner_bag.count(name) / total_entries
                actual_percentage = (wins / num_simulations) * 100
                print(f"{name:<15} 模拟中奖次数: {wins:<5} ({actual_percentage:.2f}%) "
                      f"理论概率: ({theoretical_probability:.2%})")
            print("----------------")
    else:
        print("无法进行抽奖，因为没有有效参与者。")