转盘指定概率抽奖

前段时间项目中开了一个转盘抽奖的活动,转盘抽奖是一个比较简单的算法,下面介绍两种实现

第一种实现

import random
from typing import List

def random_pick(item_list: List[str], probability_list: List[float]) -> str
    """
    以设定的概率从列表里随机抽取一个值返回
    :param item_list: 元素列表
    :param probability_list: 概率列表
    :return: 抽取的元素
    """
    if not all([item_list, probability_list]):
        # 如果传入空列表,返回提示信息
        return "元素列表和概率列表不能为空!"
    x = random.uniform(0, 1)        # 产生随机数
    accumulative_probability = 0.0  # 累积概率
    for item, probability in zip(item_list, probability_list):
        accumulative_probability += probability
        if x < accumulative_probability:
            break
    return item


### 第二种实现 ```python import random

def random_pick():
lucky_spin_rewards = [
{"integral": 1, "probability": 20},
{"integral": 2, "probability": 10},
{"integral": 3, "probability": 60},
{"integral": 4, "probability": 10},
]
integral = 0
x = random.randint(1, 100)
for item in lucky_spin_rewards:
probability = item["probability"]
if x <= probability:
integral = item["integral"]
break
x -= probability

return integral

以上两种实现其实原理一样,只是顺序不一样而已
posted @ 2020-02-18 18:52  乘月归  阅读(1063)  评论(0编辑  收藏  举报