在Python中生成具有三个不同长度字母的随机字符串

Question

如何生成一个包含三个整数a、B和C的随机字符串，其中正好包含字母“a”、字母“B”和字母“C”？你知道吗

def generate_string(A, B, C):
    // return a string

s = generate_string(2, 5, 1)
// outputs can be
aabbbbbc 
ababbbcb
abcabbbb
and so on....

但是，我可以使用python随机选择（）函数。你知道吗

>>> print(''.join(choice('a'+'b'+'c') for i in range(5)))
aacaa

Answer 1

一种方法是生成一个包含这些字符的字符串，然后从该字符串^{}将其洗牌：

from random import sample

s = 'a'*2 + 'b'*5 + 'c'*1

print(''.join(sample(s, k=len(s))))

根据您的评论，函数可能类似于

from itertools import repeat, chain
from random import shuffle

def generate_string(*pairs):
    chars = list(chain.from_iterable(repeat(c, n) for n, c in pairs))
    shuffle(chars)
    return ''.join(chars)

print(generate_string((2, 'a'), (5, 'b'), (1, 'c')))

如果你只担心这3个字符，这可以简化为

def generate_string(a, b, c):
    s = 'a'*a + 'b'*b + 'c'*c
    return ''.join(sample(s, k=len(s)))

Answer 2

试试这个：

from random import choice

options = {'a': 2, 'b': 5, 'c': 1}
gen_chars = []
while options:
    genc = choice(''.join(options.keys()))
    options[genc] -= 1
    if options[genc] == 0:
        options.pop(genc)
    gen_chars.append(genc)

print(''.join(gen_chars))

Answer 3

复杂性是O（a+b+c），但这会起作用：

from random import shuffle

a = 3
b = 4
c = 5
letters = []

for i in range(a):
  letters.append('a')
for i in range(b):
  letters.append('b')
for i in range(c):
  letters.append('c')

shuffle(letters)
print(''.join(letters))