Итерация по массиву numpy

Есть ли менее многословная альтернатива этому:

for x in xrange(array.shape[0]):
    for y in xrange(array.shape[1]):
        do_stuff(x, y)

Я придумал следующее:

for x, y in itertools.product(map(xrange, array.shape)):
    do_stuff(x, y)

Что сохраняет один отступ, но все равно довольно некрасиво.

Я надеюсь на что-то похожее на этот псевдокод:

for x, y in array.indices:
    do_stuff(x, y)

Есть ли что-нибудь подобное?

127
задан Ram Rachum 6 August 2011 в 14:27
поделиться