combinatorics via __future__ generators

[Phlip]

Python:

I have a quaint combinatorics problem. Before I solve it, or find a
solution among "generators", I thought y'all might like to show off
any solutions.

Given an array like this...

[0, 4, 3]

Produce an array like this:

[
[0, 0, 0],
[0, 1, 0],
[0, 2, 0],
[0, 3, 0],
[0, 1, 1],
[0, 2, 1],
[0, 3, 1],
[0, 1, 2],
[0, 2, 2],
[0, 3, 2],
]

The first array is the counts of options in 4 slots, and the second is
all combinations of indexes of each option, such that every option
associates once with every other option. The leading 0 simply
represents a slot with no options; the algorithm must preserve those.

This should be child's play for the generator package, right?

 

[Phlip]

Python:

I have a quaint combinatorics problem. Before I solve it, or find a
solution among "generators", I thought y'all might like to show off
any solutions.

Given an array like this...

  [0, 4, 3]

Produce an array like this:

  [
    [0, 0, 0],
    [0, 1, 0],
    [0, 2, 0],
    [0, 3, 0], [0, 0, 1],
[0, 1, 1],
    [0, 2, 1],
    [0, 3, 1], [0, 0, 2],
[0, 1, 2],
    [0, 2, 2],
    [0, 3, 2],
]

I forgot those combinations!

 

[Chris Rebert]

Python:

I have a quaint combinatorics problem. Before I solve it, or find a
solution among "generators", I thought y'all might like to show off
any solutions.

Given an array like this...

 [0, 4, 3]

Produce an array like this:

 [
   [0, 0, 0],
   [0, 1, 0],
   [0, 2, 0],
   [0, 3, 0],
   [0, 1, 1],
   [0, 2, 1],
   [0, 3, 1],
   [0, 1, 2],
   [0, 2, 2],
   [0, 3, 2],
]

The first array is the counts of options in 4 slots, and the second is
all combinations of indexes of each option, such that every option
associates once with every other option. The leading 0 simply
represents a slot with no options; the algorithm must preserve those.

This should be child's play for the generator package, right?

from itertools import imap, product

def special_range(n):
return (xrange(n) if n else [0])

def something(arr):
return product(*imap(special_range, arr))

print list(something([0,4,3]))

Cheers,
Chris

 

[Phlip]

Awesome thanks - but:

from itertools import imap,product

Do we have a version for Python2.5? I have to support an older server
here; can't install a newer python on it...

 

[John Yeung]

Awesome thanks - but:


Do we have a version for Python2.5? I have to support an older server
here; can't install a newer python on it...

If you can get by with the performance of pure Python, a solution is
right in the documentation for 2.6:

###
def product(*args, **kwds):
# product('ABCD', 'xy') --> Ax Ay Bx By Cx Cy Dx Dy
# product(range(2), repeat=3) --> 000 001 010 011 100 101 110 111
pools = map(tuple, args) * kwds.get('repeat', 1)
result = [[]]
for pool in pools:
result = [x+[y] for x in result for y in pool]
for prod in result:
yield tuple(prod)
###

The docs for itertools are full of these definitions (for 2.6
functions) that can be used as recipes (if you don't have 2.6).

John