简单明了的均匀概率分布的生成函数

假设现在有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不一样就可以实现不同概率区间分布了