此文做了两段很简单的代码,以展示使用不同思路,都能达到相同效果,但执行效率和代码可读性差别比较大,只希抛砖引玉。
某企业有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()方法,直接批量生成中奖员工列表,而不需要再一个一个生成中奖员工,代码的可读性及效率都有所提升。