numpy.ndindex#
- class numpy.ndindex(*shape)[source]#
用于索引数组的 N 维迭代器对象。
给定数组的形状,
ndindex
实例会迭代数组的 N 维索引。每次迭代都会返回一个索引元组,最后一个维度优先迭代。- 参数:
- shape整数,或单个整数元组
数组每个维度的尺寸可以作为单独的参数传递,也可以作为元组的元素传递。
参见
示例
>>> 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)
方法
ndincr
()将多维索引加一。