計算 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 矩陣的內容。請記住，此方法僅適用於方形矩陣，如果我們在矩形矩陣上嘗試，則會出現錯誤。