numpy.nested_iters

numpy.nested_iters()

创建要在嵌套循环中使用的指示器

创建的元组 nditer 对象,在op参数的不同轴上迭代嵌套循环。第一个迭代器用于最外层的循环,最后一个用于最内层的循环。前进一个将改变后续的迭代器以指向它的新元素。

参数
op数组的数组或序列

要循环访问的数组。

axesint列表

每个项目都用作一个Nditer的“op_axes”参数。

flags, op_flags, op_dtypes, order, casting, buffersize (optional)

nditer 同名参数

返回
iters无损检测元组

中每个项目的一个指示器 axes ,最外面的第一个

参见

nditer

实例

基本用法。注意“扁平”版本的 [A:,0,,] ,A [,1, 0] ,A [,2,] ]因为我们把第一个ITER的轴指定为 [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