[Numpy-discussion] how do I list all combinations

Charles R Harris charlesr.harris@gmail....
Wed Dec 26 15:30:43 CST 2007


On Dec 26, 2007 1:45 PM, Keith Goodman <kwgoodman@gmail.com> wrote:

> On Dec 26, 2007 12:22 PM, Mathew Yeates <myeates@jpl.nasa.gov> wrote:
> > I have an arbitrary number of lists. I want to form all possible
> > combinations from all lists. So if
> > r1=["dog","cat"]
> > r2=[1,2]
> >
> > I want to return [["dog",1],["dog",2],["cat",1],["cat",2]]
> >
> > It's obvious when the number of lists is not arbitrary. But what if
> > thats not known until runtime?
>
> Would this work?
>
> Make a function that takes two inputs (a list of lists and a list) and
> returns a list of lists that contains all possible combinations.
> Iterate through all lists by calling the function with the output of
> the previous call (a list of lists) and the next list.
> ____
>

Yeah, you can do it with recursion, but I don't think it would be quite as
efficient. An example of the explicit approach, define the following
generator:

def count(listoflists) :
    counter = [i[0] for i in listoflists]
    maxdigit = [len(i) - 1 for i in listoflists]
    yield counter
    while True :
        i = 0;
        while i < len(counter) and counter[i] == maxdigit[i] :
            counter[i] = 0
            i += 1
        if i < len(counter) :
            counter[i] += 1
            yield counter
        else :
            return



In [30]: a = ['dog', 'cat', 'horse']

In [31]: b = ['red', 'black']

In [32]: c = [a,b]

In [33]: for i in count.count(c) : print [c[j][i[j]] for j in range(len(c))]
   ....:
['dog', 'red']
['cat', 'red']
['horse', 'red']
['dog', 'black']
['cat', 'black']
['horse', 'black']

___________________________________________
> Numpy-discussion mailing list
> Numpy-discussion@scipy.org
> http://projects.scipy.org/mailman/listinfo/numpy-discussion
>
-------------- next part --------------
An HTML attachment was scrubbed...
URL: http://projects.scipy.org/pipermail/numpy-discussion/attachments/20071226/5806773c/attachment-0001.html 


More information about the Numpy-discussion mailing list