Python 实现年会抽奖

import pandas as pd
import random

df = pd.read_excel('D:\年会座位安排.xlsx')

all_numbers = list(range(500))

# 随机选取 50 个数，三等奖
selected_50 = random.sample(all_numbers, 50)

# 从 Excel 中获取对应序号的第一列（工号）
selected_50_work_numbers = df.iloc[selected_50, 0].values

# 从没有被选取的数中取 20 个，二等奖
remaining_numbers = list(set(all_numbers) - set(selected_50))
selected_20 = random.sample(remaining_numbers, 20)
selected_20_work_numbers = df.iloc[selected_20, 0].values

# 一等奖
last_remaining_numbers = list(set(remaining_numbers) - set(selected_20))
selected_10 = random.sample(last_remaining_numbers, 10)
selected_10_work_numbers = df.iloc[selected_10, 0].values

print("三等奖：", sorted(selected_50_work_numbers))  # 对工号排序，便于员工查找
print("二等奖：", sorted(selected_20_work_numbers))
print("一等奖：", sorted(selected_10_work_numbers))