Python 按比例获取样本数据或执行任务

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


__author__ = 'shouke'

import time
from copy import deepcopy


def main():

    class_propotion_map =  {'A':3, 'B':5, 'C':7, 'D':7} # 分类及样本数比例投射
    class_list = [] # 分类
    class_proption_list = [] # 储放分类样本数比例

    for class_type, propotion in class_propotion_map.items(): # 同一个循环系统，能够确保比例数据库索引和相匹配分类数据库索引一一对应
        class_list.append(class_type) 
        class_proption_list.append(propotion)

    temp_class_propotion_list = deepcopy(class_proption_list)
    result = []

    t1 = time.time()
    total_sample_num = 100000 #每日任务执行次数
    for i in range(1, total_sample_num 1):
        max_propotion = max(temp_class_propotion_list)
        if max_propotion > 0:
            index = temp_class_propotion_list.index(max_propotion)
            result.append(class_list[index])
            temp_class_propotion_list[index] -= 1
        elif max_propotion == 0 and min(temp_class_propotion_list) == 0:
            temp_class_propotion_list = deepcopy(class_proption_list)
            index = temp_class_propotion_list.index(max(temp_class_propotion_list))
            result.append(class_list[index])
            temp_class_propotion_list[index] -= 1



    t2 = time.time()
    from collections import Counter
    c = Counter(result)
    for item in c.items():
        print(item[0], item[1]/total_sample_num)
    print('用时：%s'%(t2-t1))

main()