python抽奖代码

turntable.py

# turntable.py 幸运转盘抽奖
import random
import bisect

class WeightRandom:
    def __init__(self, items):
        weights = [w for _, w in items]
        self.goods = [x for x, _ in items]
        self.total = sum(weights)
        self.acc = list(self.accumulate(weights))

    def accumulate(self, weights):  # 累和.如accumulate([10,40,50])->[10,50,100]
        cur = 0
        for w in weights:
            cur = cur + w
            yield cur

    def __call__(self):
        return self.goods[bisect.bisect_right(self.acc, random.uniform(0, self.total))]

# 返回list中tuple的索引为0的元素，这里就是iphone、ipad、mac
prizename = WeightRandom([('iphone',20),('ipad',30),('mac',50)])
print(prizename)