python coordinates generator

Question

I need to generate all points coordinates by given dimensions, for example:

>>>points(2,2)
>>>[(0,0),(0,1),(1,0),(1,1)]
>>>points(1,1,1)
>>>[(0,0,0),(0,0,1),(0,1,0),(0,1,1),(1,1,0),(1,1,1)]

I,ve seen solution for 2-dimensions, but cant find a way to make method independent of number of dimensions:

>>> from itertools import product
>>> list(product(xrange(d) for d in (1,2,3)))
[(xrange(1),), (xrange(2),), (xrange(3),)] #where is my tuples?

where (1,2,3) are *args tuple, which can be anything.

svk · Accepted Answer · 2013-03-10 18:23:21Z

2

You need to use a * when calling the function to use an iterable object as an argument list. f(*[1,2,3]) works like f(1,2,3).

Knowing this, and using itertools:

def points(*args):
  return list(product(*[range(n) for n in args]))

answered Mar 10, 2013 at 18:23

svk

5,94920 silver badges22 bronze badges

Sign up to request clarification or add additional context in comments.

1 Answer 1