python两种抽奖代码对比

此文做了两段很简单的代码,以展示使用不同思路,都能达到相同效果,但执行效率和代码可读性差别比较大,只希抛砖引玉。

要求:

某企业有300名员工,开年会需要抽奖,奖项如下:
一等奖:3名
二等奖:6名
三等奖:30名
共抽三次,先抽三等奖,再抽二等奖,最后抽一等奖。每个员工只能中奖一次,不能重复。

代码一:

import random
person = 300
alllist = []
for i in range(1,person + 1) :
    alllist.append('per'+str(i))
jiang = ['三等奖','二等奖','一等奖']
no = 0
for i in [30,6,3] :
    print(f'正在抽取{jiang[no]}。')
    for b in range(i) :
        while True :
            num = 'per' + str(random.randrange(person))  # 随机生成中奖员工编号
            
            if num in alllist :  #如果随机生成的员工未被删除
                break
            else :
                continue
        print(f'第{b+1}名{jiang[no]}是{num}')
        alllist.remove(num)  # 删除已经中奖的员工
    no += 1

代码二

import random
person = 300
alllist = []
for i in range(1,person + 1) :
    alllist.append('per'+str(i))
jiang = ['三等奖','二等奖','一等奖']
getpernum = [30,6,3]
no = 0

for i in getpernum :
    getper = random.sample(alllist, i)  #直接生成i个中奖员工的列表
    print(f'获得{jiang[no]}的有:',getper)
    for num in getper :
        alllist.remove(num)  #循环删除已经中奖的员工
    print(f'还剩下{len(alllist)}人')  #此句为调试用。
    person -= getpernum[no]
    no += 1

代码二的优点在于:使用random模块的sample()方法,直接批量生成中奖员工列表,而不需要再一个一个生成中奖员工,代码的可读性及效率都有所提升。

猜你喜欢

转载自blog.csdn.net/any1where/article/details/128554657