利用python编写年会抽奖小程序

如有不足，还请大神指点~

需求：

1. 参与抽奖的员工名单从cvs文件导入

2. 一共5轮抽奖：幸运奖两轮（15名*2），三等奖（6名），二等奖（3名），一等奖（1名）

3. 输入“s”开始抽奖

4. 在上一轮被抽中者要被从下一轮的名单中剔除

补充需求：

1. 如有未出席者或其他原因需要补抽的情况下，可以随时在当轮补抽

代码：

# -*- coding: utf-8 -*-

'''

作者：Nicole

版本：2.0

日期：2019年1月25日

项目名称：年会抽奖小程序

'''

import pandas as pd

import random

#导入员工列表

staff_list = pd.read_csv(r'/Users/nicole/Desktop/抽奖名单.csv')

list_round =list(staff_list['工号'])

prize_list = []

n_prize = [15,15,6,3,1]

name_prize = ['第一轮幸运奖','第二轮幸运奖','三等奖','二等奖','一等奖']

#剔除已经中奖人员的函数

def new_list_generate(original_list,prize_list):

    new_list =[]

    for i in original_list:

        if i not in prize_list:

            new_list.append(i)

    return new_list

#抽奖函数

def lottary_start_main(list_round,prize_list):

    v =0

    while v <5:

        start = input('输入"s"开始：')

            if start in ['s','S']:

                list_round = new_list_generate(list_round,prize_list)

                prize_list = random.sample(list_round,n_prize[v])

                print(name_prize[v],'：',prize_list)           

                v = v+1

            else:

                print('输入错误...')

if __name__ =='__main__':

    lottary_start_main(list_round,prize_list)

总结：

1. 补充需求还没做，等完善了再更新

2. 提高某些员工的中奖概率，下次写

3. 简书怎么贴代码？