python - python中两个列表的排列

Question

我有两个列表，例如:

list1 = ['square','circle','triangle']
list2 = ['red','green']

如何创建这些列表的所有排列，如下所示:

[
  'squarered', 'squaregreen',
  'redsquare', 'greensquare',
  'circlered', 'circlegreen',
  'redcircle', 'greencircle',
  'trianglered', 'trianglegreen',
  'redtriangle', 'greentriangle'
]

我可以为此使用 itertools 吗？

Answer 1

你想要 itertools.product 方法，它将为您提供 Cartesian product 两个列表。

>>> import itertools
>>> a = ['foo', 'bar', 'baz']
>>> b = ['x', 'y', 'z', 'w']

>>> for r in itertools.product(a, b): print r[0] + r[1]
foox
fooy
fooz
foow
barx
bary
barz
barw
bazx
bazy
bazz
bazw

您的示例要求双向乘积(即，您既想要“​​xfoo”又想要“foox”)。要做到这一点，只需做另一个产品并将结果链接起来:

>>> for r in itertools.chain(itertools.product(a, b), itertools.product(b, a)):
...   print r[0] + r[1]