numpy.indices

原文：https://docs.scipy.org/doc/numpy/reference/generated/numpy.indices.html

译者：飞龙 UsyiyiCN

校对：（虚位以待）

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

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

计算数组，其中子数组包含仅沿对应轴变化的索引值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]直接提取所需的元素更为直接。