需求
遇到一些情况需要指定概率的条件下生成随机数,比如要生成姓名字符串,随机生成的话一般是两个字的姓名和3个字的姓名概率是一样的,而实际应该是大部分偏向三个字的,所以需要设置3和2的随机比例,比如3个字的占80%,2个字的占20%,找了一圈好像有看到numpy来处理,但是因为个人程序要打包,所以基本放弃了这个,就自己基于random写了个函数
解决方法
代码如下:
def p_random(arr1,arr2):
assert len(arr1) == len(arr2), "Length does not match."
assert sum(arr2) == 1 , "Total rate is not 1."
sup_list =