计算 NumPy 矩阵的幂

本文将介绍如何在 NumPy 中计算矩阵的幂。

使用 numpy.linalg.matrix_power() 函数计算 NumPy 矩阵的幂

numpy.linalg 库中的 matrix_power() 函数 用于计算矩阵的幂。它将矩阵和指数作为输入参数，并在另一个矩阵中返回运算结果。请参考以下代码示例。

import numpy as np

matrix = np.array([[1, 2], [4, 5]])

power = np.linalg.matrix_power(matrix, 3)
print(power)

输出：

[[ 57  78]
 [156 213]]

我们首先使用上面代码中的 np.array() 函数将矩阵创建为 2D NumPy 数组。然后我们使用 matrix_power() 函数计算矩阵的立方并将结果存储在 power 矩阵中。最后，我们显示了 power 矩阵的内容。请记住，此方法仅适用于方形矩阵，如果我们在矩形矩阵上尝试，则会出现错误。