Python实现基于CSV抽奖券的随机中奖者选择教程

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

import csv
import random
from collections import Counter # 用于后续验证

def read_raffle_data(filepath: str) -> list[tuple[str, int]]:
    """
    从CSV文件读取抽奖参与者及其抽奖券数量。
    假设CSV文件第一行为标题，包含'Name'和'Ticket count'。
    """
    participants_data = []
    try:
        with open(filepath, newline="", encoding="utf-8") as f:
            reader = csv.reader(f, skipinitialspace=True)
            next(reader)  # 跳过标题行

            for row in reader:
                if len(row) == 2:
                    name = row[0].strip()
                    ticket_count = int(row[1].strip())
                    if ticket_count > 0: # 确保票数大于0
                        participants_data.append((name, ticket_count))
                else:
                    print(f"警告: 跳过格式不正确的行: {row}")
    except FileNotFoundError:
        print(f"错误: 文件未找到 - {filepath}")
    except ValueError as e:
        print(f"错误: 转换抽奖券数量时出错，请检查CSV文件格式: {e}")
    return participants_data

# 示例使用
# participants = read_raffle_data("raffle.csv")
# print(participants) # 例如: [('Alice', 3), ('Bob', 2), ...]

def create_bag_of_names(participants_data: list[tuple[str, int]]) -> list[str]:
    """
    根据参与者数据创建“抽奖券袋”列表。
    每个名字根据其票数重复出现。
    """
    bag_of_names: list[str] = []
    for name, ticket_ct in participants_data:
        bag_of_names.extend([name] * ticket_ct)
    return bag_of_names

# 示例使用
# participants = [('Alice', 3), ('Bob', 2), ('Charlie', 4), ('Dana', 1)]
# bag = create_bag_of_names(participants)
# print(bag) 
# 预期输出: ['Alice', 'Alice', 'Alice', 'Bob', 'Bob', 'Charlie', 'Charlie', 'Charlie', 'Charlie', 'Dana']

def pick_winner(bag_of_names: list[str]) -> str:
    """
    从“抽奖券袋”中随机选择一位中奖者。
    """
    if not bag_of_names:
        return "没有可供抽奖的参与者。"
    return random.choice(bag_of_names)

# 示例使用
# winner = pick_winner(bag)
# print(f"本次抽奖的中奖者是: {winner}")

import csv
import random
from collections import Counter

def read_raffle_data(filepath: str) -> list[tuple[str, int]]:
    """
    从CSV文件读取抽奖参与者及其抽奖券数量。
    假设CSV文件第一行为标题，包含'Name'和'Ticket count'。
    """
    participants_data = []
    try:
        with open(filepath, newline="", encoding="utf-8") as f:
            reader = csv.reader(f, skipinitialspace=True)
            next(reader)  # 跳过标题行

            for row in reader:
                if len(row) == 2:
                    name = row[0].strip()
                    try:
                        ticket_count = int(row[1].strip())
                        if ticket_count > 0:
                            participants_data.append((name, ticket_count))
                        else:
                            print(f"警告: 参与者 '{name}' 的抽奖券数量为非正数 ({ticket_count})，已忽略。")
                    except ValueError:
                        print(f"警告: 参与者 '{name}' 的抽奖券数量 '{row[1]}' 无效，已忽略。")
                else:
                    print(f"警告: 跳过格式不正确的行: {row}")
    except FileNotFoundError:
        print(f"错误: 文件未找到 - {filepath}")
    except Exception as e:
        print(f"读取CSV文件时发生意外错误: {e}")
    return participants_data

def create_bag_of_names(participants_data: list[tuple[str, int]]) -> list[str]:
    """
    根据参与者数据创建“抽奖券袋”列表。
    每个名字根据其票数重复出现。
    """
    bag_of_names: list[str] = []
    for name, ticket_ct in participants_data:
        bag_of_names.extend([name] * ticket_ct)
    return bag_of_names

def conduct_raffle(filepath: str) -> str:
    """
    执行一次完整的抽奖过程并返回中奖者。
    """
    participants = read_raffle_data(filepath)
    if not participants:
        return "未能读取有效的参与者数据，无法进行抽奖。"

    bag = create_bag_of_names(participants)
    if not bag:
        return "抽奖券袋为空，无法进行抽奖。"

    winner = random.choice(bag)
    return winner

# --- 主程序入口 ---
if __name__ == "__main__":
    csv_file = "raffle.csv" # 确保此文件存在于脚本同目录下或提供完整路径

    # 第一次抽奖
    print(f"从 '{csv_file}' 进行一次抽奖...")
    winner = conduct_raffle(csv_file)
    print(f"恭喜！本次中奖者是: {winner}")
    print("-" * 30)

    # 验证抽奖公平性 (可选，用于测试)
    print("进行1000次模拟抽奖以验证公平性...")
    all_participants = read_raffle_data(csv_file)
    if all_participants:
        raffle_bag = create_bag_of_names(all_participants)
        if raffle_bag:
            num_simulations = 1000
            results = Counter()
            for _ in range(num_simulations):
                results[random.choice(raffle_bag)] += 1

            total_tickets = len(raffle_bag)
            print("模拟抽奖结果分布:")
            for name, wins in results.most_common():
                expected_percentage = (all_participants[[p[0] for p in all_participants].index(name)][1] / total_tickets) * 100
                actual_percentage = (wins / num_simulations) * 100
                print(f"{name:<10} 实际中奖次数: {wins:<5} 实际占比: {actual_percentage:.2f}% (预期占比: {expected_percentage:.2f}%)")
        else:
            print("无法进行模拟抽奖，抽奖券袋为空。")
    else:
        print("无法进行模拟抽奖，未能读取有效的参与者数据。")