之前有个类似的帖子:python按概率生成随机数
各位可以先看下,不过我没看懂他这么写的目的,好像是统计10000次生成红绿蓝色的情况分别是多少次,代码运行没什么问题,就是有时候会产生随机结果没有选中红绿蓝,这一点可以在最后结果加起来不等于10000看出来。不知道他要的就是这个效果还是有误。于是我修改了下,下面是我修改后的代码,保证了每次循环都能选中红绿蓝中的一个。
# coding: utf-8
import random
import time
def random_index(rate):
"""随机变量的概率函数"""
# 参数rate为list<int>
# 返回概率事件的下标索引
start = 0
index = 0
randnum = random.randint(1, sum(rate))
for index, scope in enumerate(rate):
start += scope
if randnum <= start:
break
return index
def main():
arr = ['red', 'green', 'blue']
rate = [45, 30, 25]
red_times = 0
green_times = 0
blue_times = 0
for i in range(10000):
index = random_index(rate)
if index == 0:
red_times += 1
elif index == 1:
green_times += 1
elif index == 2:
blue_times += 1
dict = {arr[0]: red_times, arr[1]: green_times, arr[2]: blue_times}
print(dict)
# print(red_times, green_times, blue_times)
start = time.perf_counter()
main()
end = time.perf_counter()
print(end - start)
打印结果:
{'red': 4493, 'green': 2922, 'blue': 2585}
0.026857719000000002
我也测了下之前他代码运行时间:
4513 3055 2447
0.068724921
可以看出来这个还是比较耗时间的
当然,按概率生成随机数还有更好的方法,希望大牛不吝赐教。谢谢