网上看到了各种权重 巴拉巴拉巴拉,自己写 然后你又要巴拉巴拉巴拉的看,不好用,一个random.choices 就能解决的事情为啥 要那么麻烦
上代码
nation_data = {'汉': 99, '蒙古': 1, '回': 1, '藏': 1, '维吾尔': 1}
def get_random_weight(data_dict, num=1):
"""根据权重获取key"""
key_list = []
value_list = []
for key, value in data_dict.items():
key_list.append(key)
value_list.append(value)
random_data = random.choices(key_list, weights=value_list, k=num)
return random_data
random.choices 加权 秒懂图