A weighted version of random.choice

后端 未结 25 2384
闹比i
闹比i 2020-11-21 06:29

I needed to write a weighted version of random.choice (each element in the list has a different probability for being selected). This is what I came up with:



        
25条回答
  •  轮回少年
    2020-11-21 06:58

    A very basic and easy approach for a weighted choice is the following:

    np.random.choice(['A', 'B', 'C'], p=[0.3, 0.4, 0.3])
    

提交回复
热议问题