Python中的 numpy.diag_indices()

numpy.diag_indices()是numpy中的一个函数，用于返回一个二维数组的对角线索引。

语法

numpy.diag_indices(n, ndim=2)

参数

• n：整数，可选参数，表示返回二维数组的行数和列数，即对角线的长度。
• ndim：整数，可选参数，表示返回的数组的维度数。

返回值

numpy.diag_indices()函数将返回一个由索引元组组成的二元组，其中第一个元素是由行索引和列索引组成的元组，第二个元素是一个整数数组，表示返回数组的形状。

示例

import numpy as np

# 返回3x3数组的对角线索引
diag_indices = np.diag_indices(3)

print(diag_indices)

输出结果如下：

(array([0, 1, 2]), array([0, 1, 2]))

以上代码中，diag_indices变量存储了一个二元组，其中第一个元素为 (array([0, 1, 2]), array([0, 1, 2]))。这个元组里的两个数组组成了一个3x3的二维数组的对角线索引。

import numpy as np

# 返回4x4数组的对角线索引
diag_indices = np.diag_indices(4)

print(diag_indices)

输出结果如下：

(array([0, 1, 2, 3]), array([0, 1, 2, 3]))

以上代码中，diag_indices变量存储了一个二元组，其中第一个元素为 (array([0, 1, 2, 3]), array([0, 1, 2, 3]))。这个元组里的两个数组组成了一个4x4的二维数组的对角线索引。

结论

numpy.diag_indices()是一个方便的函数，可以帮助我们快速生成二维数组的对角线索引，方便进行二维数组的操作。