我有两个角色,例如:
a = 'a'
b = 'b'
我需要找到这两个字符的所有可能组合,从而形成长度为 N 的字符串。例如,如果N = 3
:
combinations = ['aab', 'aba', 'baa', 'abb', 'bab', 'bba', 'aaa', 'bbb']
某些模块中有此功能吗?
itertools.product https://docs.python.org/3/library/itertools.html#itertools.product就是你想要的:
>>> from itertools import product
>>> a = 'a'
>>> b = 'b'
>>> N = 3
>>> lst = [a, b]
>>> [''.join(x) for x in product(lst, repeat = N)]
['aaa', 'aab', 'aba', 'abb', 'baa', 'bab', 'bba', 'bbb']
也可以用三重嵌套列表理解来编写:
>>> [x + y + z for x in lst for y in lst for z in lst]
['aaa', 'aab', 'aba', 'abb', 'baa', 'bab', 'bba', 'bbb']
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)