Python中是否有一个函数可以生成给定字母表中所有长度为n的字符串?

时间:2011-02-28 12:11:53

标签: python algorithm string

我需要一个函数generateAllStrings(n, alphabet)来做这样的事情:

generateAllStrings(4, ['a','b'])
>>> ["aaaa", "aaab", "aaba", "aabb", "abaa", .... , "bbba", "bbbb"]

换句话说,generateAllStrings(n, alphabet)应该在列表字母表中的字符上返回长度为n的所有可能的字符串。

在itertools中有这样的功能吗?

1 个答案:

答案 0 :(得分:38)

>>> [''.join(i) for i in itertools.product("ab",repeat=4)]
['aaaa', 'aaab', 'aaba', 'aabb', 'abaa', 'abab', 'abba', 'abbb', 'baaa', 'baab', 'baba', 'babb', 'bbaa', 'bbab', 'bbba', 'bbbb']