numpy.ma.indices¶
-
numpy.ma.
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 lengthN
, 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]
直接提取所需的元素更为直接。