python - 给定比例的随机非均匀分布

Question

我有 3 个标签：“A”、“B”、“C”。

我想生成一个包含 100 个元素的随机列表，其中 60% 是“A”，30% 是“B”，10% 是“C”。

我怎样才能做到这一点？（我是python新手，希望这个问题不要太傻。）

---

编辑：我的问题与这个问题略有不同： Generate random numbers with a given (numerical) distribution

就像在评论中一样，我希望它们中有 60% 是“A”，并不是每个元素都有 60% 的概率是“A”。所以 numpy.random.choice() 不是我的解决方案。

Answer 1

你可以只排列一个列表。假设您创建了列表

x = list('A'*60 + 'B'*30 + 'C'*10)

然后，您可以像这样就地洗牌：

from random import shuffle
shuffle(x)

Answer 2

如果分布应该是均匀的，那么 A 将平均出现在 60% 的情况下，因此其他值

import random
res = []
for i in range(0, n_samples):
   r = random.random()
   if(r<=0.6): res.append(A)
   elif(r>0.7): res.append(B)
   elif(r>0.6 and r<=0.7): res.append(C)

Answer 3

如果您希望60% 是 A、30% B 和 10% C，并且您知道总共必须有 100 个元素，您可以执行以下操作：

import random

num = 100
prob_a = 0.6
prob_b = 0.3
prob_c = 0.1

As = int(num*prob_a) * 'A'
Bs = int(num*prob_b) * 'B'
Cs = int(num*prob_c) * 'C'

# create a list with 60 As, 30 Bs, and 10 Cs
chars = list(As + Bs + Cs)
random.shuffle(chars)

print("".join(chars))

那会输出类似的东西BAAAAABBCBAABABAAAACAABBAABACAACBAACBBBAAACBAAAABAAABABAAAAABBBABAABAABAACCAABABAAAAAACABBBBCABAAAAA