numpy.ndindex#

класс numpy.ndindex(*shape)[источник]#

N-мерный объект итератора для индексирования массивов.

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

Параметры:

shapeцелые числа или один кортеж целых чисел: Размер каждого измерения массива может быть передан как отдельные параметры или как элементы кортежа.

Смотрите также

ndenumerate, flatiter

Примеры

>>> import numpy as np

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

>>> 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)