假设现在有n个式神放在一个桶里(每个式神有且只有一个,要按照不同概率取出一个(放回去的,就是每次取出再放回)式神。那么可以用区间来做。
比如,A,B,C三个式神,用[0,5)、[5,8)、[8,10)表示这三个式神的概率区间,那么概率计算为:
抽取A的概率 5/10 = 0.5
抽取B的概率 3/10 = 0.3
抽取C的概率 2/10 = 0.2
理论上说明白了,代码实现一下
# coding=utf-8
import random
data_list = [5, 3, 2]
# data_list[0] = 5 代表A的区间范围[0, 5)
# data_list[1] = 3 代表B的区间范围[5, 8)
# data_list[2] = 2 代表C的区间范围[8, 10)
data_sum = 10
def Generate(_data_list, _data_sum):
_data_list_len = len(_data_list)
random_num = random.uniform(0, _data_sum) # 产生一个在[0, 10)区间的随机数
for i in xrange(_data_list_len):
if random_num < _data_list[i]: # 类似于比较random_num是不是处于[0, data_list[i])区间
return i
random_num -= _data_list[i] # 当上面判断到是大于前一个区间了,就可以把整个区间平移(向x轴负方向平移)
return _data_list_len - 1
def DrawLine(_len):
line = ['*' for i in range(_len)]
print ''.join(line)
A_len = 0
B_len = 0
C_len = 0
# 随机生成200次,画线可以看出大概的概率分布
for i in xrange(200):
index = Generate(data_list, data_sum)
if index == 0:
A_len += 1
elif index == 1:
B_len += 1
elif index == 2:
C_len += 1
else:
print 'error! index = '+str(index)
DrawLine(A_len)
DrawLine(B_len)
DrawLine(C_len)
这样只要输入的data_list和data_sum不一样就可以实现不同概率区间分布了