numpy.nested_iters

numpy.nested_iters()

Создание nditer для использования во вложенных циклах

Создаёт кортеж объектов nditer, которые итерируют во вложенных циклах по различным осям аргумента op. Первый итератор используется во внешнем цикле, последний — во внутреннем. Переход к следующему элементу изменит последующие итераторы, чтобы они указывали на новый элемент.

Параметры:	op : ndarray or sequence of array_like Массив(ы) для итерации. axes : list of list of int Каждый элемент используется в качестве аргумента «op_axes» для nditer flags, op_flags, op_dtypes, order, casting, buffersize (необязательно) См. параметры nditer с таким же именем
Возвращает:	iters : tuple of nditer nditer для каждого элемента в axes, начиная с внешнего цикла

См. также

nditer

Примеры

Базовое использование. Обратите внимание, как y является «сплющенной» версией [a[:, 0, :], a[:, 1, 0], a[:, 2, :]] поскольку мы указали оси первого итератора как [1]

>>> a = np.arange(12).reshape(2, 3, 2)
>>> i, j = np.nested_iters(a, [[1], [0, 2]], flags=["multi_index"])
>>> for x in i:
...      print(i.multi_index)
...      for y in j:
...          print('', j.multi_index, y)

(0,)

(0, 0) 0 (0, 1) 1 (1, 0) 6 (1, 1) 7

(1,)

(0, 0) 2 (0, 1) 3 (1, 0) 8 (1, 1) 9

(2,)

(0, 0) 4 (0, 1) 5 (1, 0) 10 (1, 1) 11