使用 Numpy 计算矩阵的迹是线性代数中的常见运算，可用于提取有关矩阵的重要信息。矩阵的迹定义为矩阵主对角线上元素的总和，主对角线从左上角延伸到右下角。在本文中，我们将学习使用 Python 中的 NumPy 库计算矩阵迹的各种方法。
在开始之前，我们首先导入 NumPy 库 -
import numpy as np
接下来，让我们使用 np.array 函数定义一个矩阵 -
A = np.array([[1,2,3], [4,5,6], [7,8,9]])
示例 1
为了计算该矩阵的迹，我们可以使用 NumPy 中的 np.trace 函数
立即学习“Python免费学习笔记（深入）”；
import numpy as np A = np.array([[1,2,3], [4,5,6], [7,8,9]]) trace = np.trace(A) print(trace)
输出
15
np.trace 函数采用单个参数，它是我们要计算其轨迹的矩阵。它将矩阵的迹作为标量值返回。
示例 2
或者，我们也可以使用 sum 函数计算矩阵的迹，并对主对角线上的元素进行索引 -
import numpy as np A = np.array([[1,2,3], [4,5,6], [7,8,9]]) trace = sum(A[i][i] for i in range(A.shape[0])) print(trace)
输出
15
在这里，我们使用矩阵的 shape 属性来确定其维度，并使用 for 循环迭代主对角线上的元素。
需要注意的是，矩阵的迹仅针对方阵定义，即行数和列数相同的矩阵。如果您尝试计算非方阵的迹，您将收到错误。
示例 3
除了计算矩阵的迹之外，NumPy 还提供了其他几个函数和方法来执行各种线性代数运算，例如计算矩阵的行列式、逆矩阵以及特征值和特征向量。以下是 NumPy 提供的一些最有用的线性代数函数的列表 -
np.linalg.det - 计算矩阵的行列式
np.linalg.inv - 计算矩阵的逆。
np.linalg.eig - 计算矩阵的特征值和特征向量。
np.linalg.solve - 求解由矩阵表示的线性方程组
np.linalg.lstsq - 解决线性最小二乘问题。
np.linalg.cholesky - 计算矩阵的 Cholesky 分解。
要使用这些函数，您需要导入 NumPy 的 linalg 子模块−
import numpy.linalg as LA
示例 3
例如，要使用 NumPy 计算矩阵的行列式，您可以使用以下代码 -
import numpy as np import numpy.linalg as LA A = np.array([[1,2,3], [4,5,6], [7,8,9]]) det = LA.det(A) print(det)
输出
0.0
NumPy 的线性代数函数针对性能进行了优化，使其成为大规模科学和数学计算应用程序的 ui 表。除了提供广泛的线性代数函数之外，NumPy 还提供了几个用于创建和操作矩阵和 n 数组的便捷函数，例如 np.zeros、np.ones、np.eye 和 np.diag。
示例 4
这是如何使用 np.zeros 函数创建零矩阵的示例 -
import numpy as np A = np.zeros((3,3)) # Creates a 3x3 matrix of zeros print(A)
输出
这将输出以下矩阵
[[0. 0. 0.] [0. 0. 0.] [0. 0. 0.]]
示例 5
类似地，np.ones 函数可以创建一个 1 矩阵，np.eye 函数可以创建一个单位矩阵。例如 -
import numpy as np A = np.ones((3,3)) # Creates a 3x3 matrix of ones B = np.eye(3) # Creates a 3x3 identity matrix print(A) print(B)
输出
这将输出以下矩阵。
[[1. 1. 1.] [1. 1. 1.] [1. 1. 1.]] [[1. 0. 0.] [0. 1. 0.] [0. 0. 1.]]
示例 6
最后，np.diag 函数可以从给定的列表或数组创建对角矩阵。例如 -
import numpy as np A = np.diag([1,2,3]) # Creates a diagonal matrix from the given list print(A)
输出
这将输出以下矩阵。
[[1 0 0] [0 2 0] [0 0 3]]
结论
总之，NumPy 是一个用于执行线性代数运算的强大 Python 库。其广泛的功能和方法使其成为科学和数学计算的必备工具，其优化的性能使其适合大规模应用。无论您需要计算矩阵的迹、求矩阵的逆，还是求解线性方程组，NumPy 都能提供完成工作所需的工具。
