以概率生成随机数

Question

我想知道如何生成一个随机数列表（1,2），但概率不同。 即：1的概率为0.6,2的概率为0.4。

谢谢！

Answer 1

使用random.choice

import random
random.choice([1,1,1,1,1,1,2,2,2,2])

你也可以写

data = [1]*6 + [2]*4
choice = random.choice(data)