numpy.ndindex

classnumpy.ndindex(*shape)[source]

Объект N-мерного итератора для индексации массивов.

Учитывая форму массива, экземпляр ndindex итерируется по N-мерному индексу массива. На каждой итерации возвращается кортеж индексов, при этом последняя размерность итерируется первой.

Параметры:

shapeцелые числа или кортеж целых чисел

Размер каждой размерности массива можно передать как отдельные параметры или как элементы кортежа.

См. также

ndenumerate, flatiter

Примеры

Размеры как отдельные аргументы

>>> for index in np.ndindex(3, 2, 1):
...     print(index)
(0, 0, 0)
(0, 1, 0)
(1, 0, 0)
(1, 1, 0)
(2, 0, 0)
(2, 1, 0)

Те же размеры - но в кортеже (3, 2, 1)

>>> for index in np.ndindex((3, 2, 1)):
...     print(index)
(0, 0, 0)
(0, 1, 0)
(1, 0, 0)
(1, 1, 0)
(2, 0, 0)
(2, 1, 0)

Методы

ndincr()

Увеличить многомерный индекс на единицу.