|
|
 |
|
Title: Generator for permutations, combinations, selections of a sequence
Submitter: Ulrich Hoffmann
(other recipes)
Last Updated: 2003/03/20
Version no: 1.0
Category:
Algorithms
|
|
 6 vote(s)
|
|
|
|
Description:
Permutations and combinations are often required in algorithms that do a complete search of the solution space. They are typically rather large so it's best not to compute them entirely but better to lazily generate them.
This recipe uses Python 2.2 generators to create appropriate generator objects,
that can be use for example as ranges in for loops.
Source: Text Source
__version__ = "1.0"
"""xpermutations.py
Generators for calculating a) the permutations of a sequence and
b) the combinations and selections of a number of elements from a
sequence. Uses Python 2.2 generators.
Similar solutions found also in comp.lang.python
Keywords: generator, combination, permutation, selection
See also: http://aspn.activestate.com/ASPN/Cookbook/Python/Recipe/105962
See also: http://aspn.activestate.com/ASPN/Cookbook/Python/Recipe/66463
See also: http://aspn.activestate.com/ASPN/Cookbook/Python/Recipe/66465
"""
from __future__ import generators
def xcombinations(items, n):
if n==0: yield []
else:
for i in xrange(len(items)):
for cc in xcombinations(items[:i]+items[i+1:],n-1):
yield [items[i]]+cc
def xuniqueCombinations(items, n):
if n==0: yield []
else:
for i in xrange(len(items)):
for cc in xuniqueCombinations(items[i+1:],n-1):
yield [items[i]]+cc
def xselections(items, n):
if n==0: yield []
else:
for i in xrange(len(items)):
for ss in xselections(items, n-1):
yield [items[i]]+ss
def xpermutations(items):
return xcombinations(items, len(items))
if __name__=="__main__":
print "Permutations of 'love'"
for p in xpermutations(['l','o','v','e']): print ''.join(p)
print
print "Combinations of 2 letters from 'love'"
for c in xcombinations(['l','o','v','e'],2): print ''.join(c)
print
print "Unique Combinations of 2 letters from 'love'"
for uc in xuniqueCombinations(['l','o','v','e'],2): print ''.join(uc)
print
print "Selections of 2 letters from 'love'"
for s in xselections(['l','o','v','e'],2): print ''.join(s)
print
print map(''.join, list(xpermutations('done')))
Discussion:
This recipe provides both combinations and permutations and lazily generates them. You can do arbitrary calculations on the permutation/combination items not just print them.
If you require the complete list of permutations, just use the built-in list() operator. Note that the resulting list can be huge.
All x-generators defined here yield sequences with elements from the original
sequence. Their difference is in which elements they take:
xpermutations takes all elements from the sequence, order matters.
xcombinations takes n distinct elements from the sequence, order matters.
xuniqueCombinations takes n distinct elements from the sequence, order is irrelevant.
xselections takes n elements (not necessarily distinct) from the sequence, order matters.
Note that 'distinct' means "different elements in the orginal sequence" and not
"different value", i.e.
list(xuniqueCombinations('aabb',2)) is
[['a', 'a'], ['a', 'b'], ['a', 'b'], ['a', 'b'], ['a', 'b'], ['b', 'b']]
and not
[['a', 'b']].
If your sequence has only items with unique values, you won't notice the difference (no pun intended).
|
|
Add comment
|
|
Number of comments: 7
A simple refactoring, Guy Argo, 2003/07/17
I notice that the bodies of xcombinations, xuniqueCombinations, xselections are almost identical. We could refactor as follows:
def generalized(f, items, n):
if n==0: yield []
else:
for i in xrange(len(items)):
for cc in generalized(f(items, i), n-1):
yield [items[i]]+cc
def skipIthItem(items, i):
return items[:i]+items[i+1:]
def afterIthItem(items, i):
return items[i+1:]
def keepAllItems(items, i):
return items
def xcombinations(items, n):
return generalized(skipIthItem, items, n)
def xuniqueCombinations(items, n):
return generalized(afterIthItem, items, n)
def xselections(items, n):
return generalized(keepAllItems, items, n)
Add comment
A faster xuniqueCombinations algorithm, Li Daobing, 2004/09/18
def xuniqueCombinations(items, n):
if n==0: yield []
else:
for i in xrange(len(items)-n+1):
for cc in xuniqueCombinations(items[i+1:],n-1):
yield [items[i]]+cc
Add comment
Faster permutations, Connelly Barnes, 2006/03/30
def permutations(L):
if len(L) == 1:
yield [L[0]]
elif len(L) >= 2:
(a, b) = (L[0:1], L[1:])
for p in permutations(b):
for i in range(len(p)+1):
yield b[:i] + a + b[i:]
The above is 6x faster on my Pentium P3 3.0 GHz, Python 2.4.1.
Add comment
Appendum, Connelly Barnes, 2006/03/30
I meant to say: 6x faster for computing list(permutations(range(8))).
Add comment
Err, Connelly Barnes, 2006/04/13
On the last line that should have been yield p[:i] + a + p[i:] .
Add comment
much faster..., Simone Leo, 2007/02/20
... But it only works on lists:
def permutations(L):
if len(L) <= 1:
yield L
else:
a = [L.pop(0)]
for p in permutations(L):
for i in range(len(p)+1):
yield p[:i] + a + p[i:]
Add comment
Wrong terminology, paul stadfeld, 2007/06/14
Unfortunately, it's a very poor example. The terminology is
all wrong.
"xpermutations takes all elements from the sequence, order matters."
This ought to be the Cartesian Product, but it's not (no replacement).
"xcombinations takes n distinct elements from the sequence, order
matters."
If order matters, it's a PERMUTATION, period.
"xuniqueCombinations takes n distinct elements from the sequence,
order is irrelevant."
No such thing, a Combination is unique by definition.
"xselections takes n elements (not necessarily distinct) from the
sequence, order matters."
Ah, this allows a size operator, so if size = length, we get full
Cartesian Product.
The proper terminology for the Cartesian Product and
its subsets is:
Permutations with replacement
Combinations with replacement
Permutations without replacement
Combinations without replacement
And if the functions were properly labeled, you would get:
permutation without replacement - size 4
Permutations of 'love'
love loev lvoe lveo leov levo olve olev ovle ovel oelv oevl vloe vleo
vole voel velo veol elov elvo eolv eovl evlo evol
permutation without replacement - size 2
Combinations of 2 letters from 'love'
lo lv le ol ov oe vl vo ve el eo ev
combination without replacement - size 2
Unique Combinations of 2 letters from 'love'
lo lv le ov oe ve
permutation with replacement - size 2
Selections of 2 letters from 'love'
ll lo lv le ol oo ov oe vl vo vv ve el eo ev ee
full Cartesian Product, permutations with replacement - size 4
Selections of 4 letters from 'love'
llll lllo lllv llle llol lloo llov lloe llvl llvo llvv llve llel lleo
llev llee loll lolo lolv lole lool looo loov looe lovl lovo lovv love
loel loeo loev loee lvll lvlo lvlv lvle lvol lvoo lvov lvoe lvvl lvvo
lvvv lvve lvel lveo lvev lvee lell lelo lelv lele leol leoo leov leoe
levl levo levv leve leel leeo leev leee olll ollo ollv olle olol oloo
olov oloe olvl olvo olvv olve olel oleo olev olee ooll oolo oolv oole
oool oooo ooov oooe oovl oovo oovv oove ooel ooeo ooev ooee ovll ovlo
ovlv ovle ovol ovoo ovov ovoe ovvl ovvo ovvv ovve ovel oveo ovev ovee
oell oelo oelv oele oeol oeoo oeov oeoe oevl oevo oevv oeve oeel oeeo
oeev oeee vlll vllo vllv vlle vlol vloo vlov vloe vlvl vlvo vlvv vlve
vlel vleo vlev vlee voll volo volv vole vool vooo voov vooe vovl vovo
vovv vove voel voeo voev voee vvll vvlo vvlv vvle vvol vvoo vvov vvoe
vvvl vvvo vvvv vvve vvel vveo vvev vvee vell velo velv vele veol veoo
veov veoe vevl vevo vevv veve veel veeo veev veee elll ello ellv elle
elol eloo elov eloe elvl elvo elvv elve elel eleo elev elee eoll eolo
eolv eole eool eooo eoov eooe eovl eovo eovv eove eoel eoeo eoev eoee
evll evlo evlv evle evol evoo evov evoe evvl evvo evvv evve evel eveo
evev evee eell eelo eelv eele eeol eeoo eeov eeoe eevl eevo eevv eeve
eeel eeeo eeev eeee
And Combinations with replacement seems to be missing.
Add comment
|
|
|
|
|
|
|
