转盘指定概率抽奖
前段时间项目中开了一个转盘抽奖的活动,转盘抽奖是一个比较简单的算法,下面介绍两种实现
第一种实现
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
以上两种实现其实原理一样,只是顺序不一样而已