以概率列表选择对应元素,轮盘概率选择Python实现
做法和其他人一样,搞了半天明白了,ran = random.random(),是产生0———1之间的随机浮点数,
即 如下代码,第一个随机生成的数小于0.1的概率就是 10%,若第一个数大于0.1剩下的就是概率中先拿和后拿问题
import random def select(): num_ = ['a', 'b', 'c'] #概率列表 r_ = [0.1, 0.3, 0.6] sum_ = 0 ran = random.random() for num, r in zip(num_, r_): sum_ += r if ran < sum_ :break return num """ 验证结果,多次循环把a,b,c单独出现的次数计数,判断对应概率 """ num_a, num_b ,num_c = 0, 0, 0 for each in range(100000): result = select() if result == 'a': num_a += 1 elif result == 'b': num_b += 1 else: num_c += 1 print(num_a, num_b, num_c)
结果如下还是符合预期的:
9983 29973 60044