从具有不同列表长度的列表中获取加权随机值

import random

def get_weighted_random_value(data):
    total_weight = sum(data.values())
    random_num = random.uniform(0, total_weight)
    cumulative_weight = 0
    
    for key, weight in data.items():
        cumulative_weight += weight
        if cumulative_weight >= random_num:
            return key

# 示例数据
data1 = {'A': 1, 'B': 2, 'C': 3}
data2 = {'X': 5, 'Y': 10, 'Z': 15, 'W': 20}

# 获取加权随机值
result1 = get_weighted_random_value(data1)
result2 = get_weighted_random_value(data2)

print(result1)  # 输出：B、C、A的概率分别为2/6、3/6、1/6
print(result2)  # 输出：W、Z、Y、X的概率分别为20/50、15/50、10/50、5/50