CODEWITHSUNDEEP
python
Edward
Edward

Reputation: 967

permutations of two lists in python

I have two lists like:

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

How can I create all permutations of these lists, like this:

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

Can I use itertools for this?

Upvotes: 62

Views: 86560

Answers (6)

Kiran Krishnan
Kiran Krishnan

Reputation: 31

Simple list comprehension will give the result

from itertools import product
final_list = [a + b for a, b in product(list1, list2)]

Upvotes: -1

John Feminella
John Feminella

Reputation: 311745

You want the itertools.product method, which will give you the Cartesian product of both lists.

>>> 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

Your example asks for the bidirectional product (that is, you want 'xfoo' as well as 'foox'). To get that, just do another product and chain the results:

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

Upvotes: 114

Nadia Alramli
Nadia Alramli

Reputation: 115011

>>> 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']

Upvotes: 47

unutbu
unutbu

Reputation: 880807

I think what you are looking for is the product of two lists, not the permutations:

#!/usr/bin/env python
import itertools
list1=['square','circle','triangle'] 
list2=['red','green']
for shape,color in itertools.product(list1,list2):
    print(shape+color)

yields

squarered
squaregreen
circlered
circlegreen
trianglered
trianglegreen

If you'd like both squarered and redsquare, then you could do something like this:

for pair in itertools.product(list1,list2):
    for a,b in itertools.permutations(pair,2):
        print(a+b)

or, to make it into a list: 

l=[a+b for pair in itertools.product(list1,list2)
   for a,b in itertools.permutations(pair,2)]
print(l)

yields 

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

Upvotes: 12

extraneon
extraneon

Reputation: 23980

You can in any case do something like:

perms = []
for shape in list1:
  for color in list2:
    perms.append(shape+color)

Upvotes: 9

Michał Marczyk
Michał Marczyk

Reputation: 84379

How about

[x + y for x in list1 for y in list2] + [y + x for x in list1 for y in list2]

Example IPython interaction:

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']

Upvotes: 20

Related Questions