前言
年会时公司用的一个抽奖的软件出错了, 于是想着自己写一个吧.这类的程序用python来实现是再简单不过了.
一、创建主持人 从csv 文件中导入员工 姓名,形成员工池
import pandas as pd
import random
# 一个主持人类
class compere():
# 初始化员工
def __init__(self):
self.staff = pd.DataFrame()
# 从csv文件中导入员工名单
def buildStaffCsv(self, path):
self.staff =pd.read_csv(path)
print(self.staff.shape)
# 从员工池中抽出中奖人数
def raffle(self, num):
sequence = self.staff.index
sequence =list(sequence)
result = random.sample( sequence, num)
for no in result:
print( self.staff.loc[no])
self.staff.drop( no, inplace=True )
return result
用pandas从csv文件中读入员工信息,
二、抽奖开始,用随机数选出一次中奖的员工, 将中奖的员工从员工池剔除,如此进入一下轮抽奖
from compere import compere
if __name__ == '__main__':
mrYang = compere()
mrYang.buildStaffCsv( 'data/staff.csv');
print(" 一等奖")
firstPrize =mrYang.raffle(2)
print( firstPrize )
print(" 二等奖")
secondPrize = mrYang.raffle(3)
print( secondPrize )
print(" 三等奖")
thirdPrize =mrYang.raffle( 4)
print( thirdPrize )
总结
很简单的一个小应用,希望后面能用得着.