2021-07-21

根据权重随机选择

最近公司要做一个广告系统，要求可以控制广告分组的展示量，具体流程可以分为以下几步：

• 1.将广告素材分组，并给分组起名，分配权重。
• 2.在每次请求广告时获取所有分组的组id和权重weight，将这两项组成元组并放在列表中，并根据权重排序，获取列表长度len(list)
• 3.将所有权重值相加，并在1到sum(weight)的区间内随机生成一个整数rand_int
• 4.开启一个while循环，当rand_int小于sum(weight)时，将此时的列表索引赋予temp_index，并将列表从尾部缩短短一位，重复第4步，知道条件不成立，此时的temp_index即是最终的index，可以由此确定根据权重随机选取的广告组。

python3代码如下

sql = f"select `weight`, `id` from {TableName.ADS_DC_PROJECT};"
weight_id_mappings = sorted(db.find_datas(sql))
weights = [item[0] for item in weight_id_mappings]
rand_num = random.randint(1, sum(weights))
flag = len(weights)
project_id = None
while rand_num <= sum(weights[:flag]):
    project_id = weight_id_mappings[flag - 1][1]
    flag -= 1

• 此代码可以实现按权重获取广告的效果，当样本量足够大时，展示结果可以获得按权重设置的比例。