numpy.indices

numpy.indices(dimensions, dtype=<type 'int'>)

返回表示网格索引的数组。

计算数组，其中子数组包含仅沿对应轴变化的索引值0,1，...。

参数：

尺寸：整数序列

网格的形状。

dtype：dtype，可选

结果的数据类型。

返回：

网格：ndarray

The array of grid indices, grid.shape = (len(dimensions),) + tuple(dimensions).

也可以看看

mgrid，meshgrid

笔记

The output shape is obtained by prepending the number of dimensions in front of the tuple of dimensions, i.e. if dimensions is a tuple (r0, ..., rN-1) of length N, the output shape is (N,r0,...,rN-1).

子阵列grid[k]包含沿着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]直接提取所需的元素更为直接。