numpy.indices

numpy.indices(dimensions, dtype=<class 'int'>) [source]

Возвращает массив, представляющий индексы сетки.

Вычисляет массив, где подмассивы содержат индексы 0,1,…, изменяющиеся только вдоль соответствующей оси.

Параметры:	dimensions : sequence of ints Форма сетки. dtype : dtype, optional Тип данных результата.
Возвращает:	grid : ndarray Массив индексов сетки, grid.shape = (len(dimensions),) + tuple(dimensions).

См. также

mgrid, meshgrid

Примечания

Форма выходных данных получается путем добавления числа измерений перед кортежем измерений, т.е. если dimensions это кортеж (r0, ..., rN-1) длины N, то форма выходных данных (N,r0,...,rN-1).

Подмассивы grid[k] содержат N-мерный массив индексов вдоль k-th оси. Явно:

grid[k,i0,i1,...,iN-1] = ik

Примеры

>>> grid = np.indices((2, 3))
>>> grid.shape
(2, 2, 3)
>>> grid[0]        # row indices
array([[0, 0, 0],
       [1, 1, 1]])
>>> grid[1]        # column indices
array([[0, 1, 2],
       [0, 1, 2]])

Индексы могут быть использованы как индексы в массиве.

>>> x = np.arange(20).reshape(5, 4)
>>> row, col = np.indices((2, 3))
>>> x[row, col]
array([[0, 1, 2],
       [4, 5, 6]])

Обратите внимание, что в приведенном выше примере было бы проще извлечь необходимые элементы напрямую с помощью x[:2, :3].