Рассмотрим, если бы у меня была функция, которая принимала аргумент кортежа (x, y), где x находилось в диапазоне (X) и y в диапазоне (Y), нормальный способ сделать это:
for x in range(X):
for y in range(Y):
function(x,y)
есть ли способ сделать
for xy in something_like_range(X,Y):
function(xy)
такое, что xy является кортежем (x, y)?
Вы можете использовать продукт из itertools
>>> from itertools import product
>>>
>>> for x,y in product(range(3), range(4)):
... print (x,y)
...
(0, 0)
(0, 1)
(0, 2)
(0, 3)
(1, 0)
(1, 1)
(1, 2)
(1, 3)
... and so on
Ваш код будет выглядеть так:
for x,y in product(range(X), range(Y)):
function(x,y)
Вы можете использовать itertools.product():
from itertools import product
for xy in product(range(X), range(Y)):
function(xy)
Pythonic → (изменяйте в соответствии с вашими требованиями)
>>> [ (x,y) for x in range(2) for y in range(2)]
[(0, 0), (0, 1), (1, 0), (1, 1)]
Версия генератора:
gen = ( (x,y) for x in range(2) for y in range(2) )
>>> for x,y in gen:
... print x,y
...
0 0
0 1
1 0
1 1
Попробуйте продукт из itertools: http://docs.python.org/library/itertools.html#itertools.product
from itertools import product
for x, y in product(range(X), range(Y)):
function(x, y)
from itertools import product
def something_like_range(*sizes):
return product(*[range(size) for size in sizes])
для использования, близкого к тому, что вы хотели:
for x,y in something_like_range(X,Y):
your_function(x,y)
=)