У меня есть два списка:
list1 = ['square','circle','triangle']
list2 = ['red','green']
Как я могу создать все перестановки этих списков, например:
[
'squarered', 'squaregreen',
'redsquare', 'greensquare',
'circlered', 'circlegreen',
'redcircle', 'greencircle',
'trianglered', 'trianglegreen',
'redtriangle', 'greentriangle'
]
Могу ли я использовать itertools для этого?
>>> import itertools
>>> map(''.join, itertools.chain(itertools.product(list1, list2), itertools.product(list2, list1)))
['squarered', 'squaregreen', 'circlered',
'circlegreen', 'trianglered', 'trianglegreen',
'redsquare', 'redcircle', 'redtriangle', 'greensquare',
'greencircle', 'greentriangle']
Вам нужен метод itertools.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]
Как насчет
[x + y for x in list1 for y in list2] + [y + x for x in list1 for y in list2]
Пример взаимодействия с IPython:
In [3]: list1 = ['square', 'circle', 'triangle']
In [4]: list2 = ['red', 'green']
In [5]: [x + y for x in list1 for y in list2] + [y + x for x in list1 for y in list2]
Out[5]:
['squarered',
'squaregreen',
'circlered',
'circlegreen',
'trianglered',
'trianglegreen',
'redsquare',
'greensquare',
'redcircle',
'greencircle',
'redtriangle',
'greentriangle']
Я думаю, что то, что вы ищете, является произведением двух списков, а не перестановок:
#!/usr/bin/env python
import itertools
list1=['square','circle','triangle']
list2=['red','green']
for shape,color in itertools.product(list1,list2):
print(shape+color)
дает
squarered
squaregreen
circlered
circlegreen
trianglered
trianglegreen
Если вы хотите как squarered, так и redsquare, вы можете сделать что-то вроде этого:
for pair in itertools.product(list1,list2):
for a,b in itertools.permutations(pair,2):
print(a+b)
или, чтобы сделать его в список:
l=[a+b for pair in itertools.product(list1,list2)
for a,b in itertools.permutations(pair,2)]
print(l)
дает
['squarered', 'redsquare', 'squaregreen', 'greensquare', 'circlered', 'redcircle', 'circlegreen', 'greencircle', 'trianglered', 'redtriangle', 'trianglegreen', 'greentriangle']
В любом случае вы можете сделать что-то вроде:
perms = []
for shape in list1:
for color in list2:
perms.append(shape+color)