Python NumPy diag方法用法及代碼示例

Numpy 的 diag(~) 方法返回輸入數組的對角線。

1. a | array-like

輸入數組。

2. k | int | optional

如果 k 為正，則返回頂部的下一個對角線。如果為負，則返回底部的對角線。默認情況下，k=0 。

輸入數組的對角線。

考慮以下二維數組：

a = np.array([[1,2,3],[4,5,6],[7,8,9]])
a



array([[1, 2, 3],
       [4, 5, 6],
       [7, 8, 9]])

獲取主對角線：

np.diag(a)



array([1, 5, 9])

要使用 k=1 獲取對角線：

np.diag(a, k=1)



array([2, 6])

要使用 k=-2 獲取對角線：

np.diag(a, k=-2)



array([7])

相關用法

注：本文由純淨天空篩選整理自Isshin Inada大神的英文原創作品 NumPy | diag method。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。