Python 按分類權重(區間)隨機獲取分類樣本

按分類權重(區間)隨機獲取分類樣本

By:授客 QQ：1033553122

開發環境

win 10

python 3.6.5

 

需求

活動抽獎，參與抽獎產品有iphone, 華爲，小米，魅族，vivo，三星手機，要求爲這些不一樣品牌的手機設置被抽獎的機率(基準機率，非絕對機率，即容許存在必定誤差)，iphone爲0，華爲0.35，小米爲0.25， 魅族0.1，vivo和三星爲0.15

 

 

代碼實現

 

#!/usr/bin/env python
# -*- coding:utf-8 -*-


__author__ = 'shouke'

import random


def get_sample_by_rate(sample_rate_list):

    if sum([item[1] for item in sample_rate_list]) != 1:
        raise ValueError("樣本比例配置錯誤，樣本佔比之和必須爲1!")

    random_normalized_num = random.random()  # random() -> x in the interval [0, 1).
    accumulated_probability = 0.0
    for sample, probabilitie in sample_rate_list:
        accumulated_probability += probabilitie
        if random_normalized_num < accumulated_probability:
            return sample

award_dict = {'iphone':0, '華爲':0.35, '小米':0.25, '魅族':0.1, 'vivo':0.15, '三星':0.15}

# 初始化
output_dict = {} # 存放取樣次數
for sample, rate in award_dict.items():
    output_dict[sample] = 0

award_list = sorted(award_dict.items(), key=lambda arg:arg[1], reverse=False)

n = 1000 # 取樣總次數
for i in range(n):
    award = get_sample_by_rate(award_list)
    output_dict[award] += 1

percentage_dict = {key: output_dict[key]/n for key in output_dict} # 存放樣本數佔比

print(output_dict)
print(percentage_dict)

　　

 

 

運行結果

 

 

 

 

注意

爲啥能夠用python的randowm函數來實現這個需求？那是由於python的random函數是平均分佈函數，產生的隨機數是等可能的。以下，能夠把[0,1)區間看做一條線，生成的隨機數能夠看做是線條上一個個點，這樣，就能夠根據這個點所在位置，把這個點劃分到某個區間（本例中劃分了幾個區間[0, 0.1)，[0.1，0.25)，[0.25，0.4)，[0.4， 0.65)，[0.65，1)），映射樣本的機率範圍

0       0.25      0.5              1

|--------|--------|----------------|

 

從運行結果來看，不難看出，這種計算方式存在必定的誤差，比較適合大數據