numpy.ndindex

class numpy.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()

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