对于四封信,我脑海中闪过的是:from random import random
def randABCD(n, pA, pB, pC, pD):
# assumes pA + pB + pC + pD == 1
cA = pA
cB = cA + pB
cC = cB + pC
def choose():
r = random()
if r < cA:
return 'A'
elif r < cB:
return 'B'
elif r < cC:
return 'C'
else:
return 'D'
return ''.join([choose() for i in xrange(n)])
我毫不怀疑这可以做得更干净/更短,我只是有点着急。在
我不满足于Dakota使用重复字符列表的答案中David的原因是,根据你的概率,可能不可能创建一个有正确数字重复的列表来模拟你想要的概率。(好吧,我想这可能永远是可能的,但是你可能需要一个巨大的列表-如果你的概率是0.11235442079,0.4072777384,0.2297927874,0.25057505341怎么办?)在
编辑:这里有一个更简洁的通用版本,可以处理任意数量的任意权重的字母:
^{pr2}$