numpy.indices()函数– Python

numpy.indices()函数返回由指定形状的数组中的每个元素的索引组成的数组。该函数返回的数组具有自定义的形状，其中每个轴的大小等于输入数组的相应轴的大小。

语法

numpy.indices(dimensions， dtype = int)

参数

• dimensions：(tuple) 所需的形状的数组
• dtype：(可选) 输出数组的数据类型

返回值

返回一个索引数组，其形状与dimensions参数相同，其中包含有从0开始产生的每个位置坐标（即索引）。

示例

下面是一个简单的示例，我们使用numpy.indices()函数创建一个2×3的二维数组，这个数组的每个元素都是它的下标。

import numpy as np

x = np.indices((2, 3))
print(x)

输出：

array([[[0, 0, 0],
        [1, 1, 1]],

       [[0, 1, 2],
        [0, 1, 2]]])

上面的代码生成了一个形状为(2, 3)的数组。该数组有两个轴，每个轴的大小为3。第一个轴的值表示行的索引，第二个轴的值表示列的索引。这个数组的所有元素都是它们自己的索引。

常见使用场景

• 根据数组形状进行切片和索引操作；
• 生成网格格式坐标数组；
• 在计算矩阵乘积时，用于调整矩阵的形状。