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).

也可以看看

mgridmeshgrid

笔记

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]直接提取所需的元素更为直接。