如何对 numpy 矩阵对角线求和 - Python

在 numpy 中，我们可以通过 numpy.trace() 函数来对矩阵的对角线求和。

具体用法如下：

import numpy as np

# 创建一个 3x3 的矩阵
matrix = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])

# 对其对角线求和
trace = np.trace(matrix)

print(trace)  # 输出结果为 15

上述代码中，我们首先使用 numpy.array() 函数创建了一个 3x3 的矩阵。然后，通过 numpy.trace() 函数对其对角线求和，并将结果保存在 trace 变量中。最后，我们通过 print() 函数将结果输出。

需要注意的是，numpy.trace() 函数只能用于求解方阵（即行数和列数相等的矩阵）的对角线和。如果我们要对一个非方阵的矩阵的对角线求和，可以使用以下代码：

import numpy as np

# 创建一个 3x4 的矩阵
matrix = np.array([[1, 2, 3, 4], [5, 6, 7, 8], [9, 10, 11, 12]])

# 获取矩阵的对角线元素
diagonal = np.diag(matrix)

# 对对角线元素求和
trace = np.sum(diagonal)

print(trace)  # 输出结果为 18

上述代码中，我们首先依然使用 numpy.array() 函数创建了一个 3x4 的矩阵。然后，通过 numpy.diag() 函数获取了该矩阵的对角线元素。最后，我们使用 numpy.sum() 函数对对角线元素求和，并将结果输出。

以上就是对 numpy 矩阵对角线求和的介绍。希望能对大家有所帮助。