以概率列表選擇對應元素,輪盤概率選擇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

 


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM