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

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

從 NumPy 1.9 開始，diagonal(~) 返回輸入數組的隻讀視圖。修改此視圖將導致錯誤。如果要修改它，請使用NumPy的copy(~)方法。

NumPy 團隊正在考慮在未來版本中將行為從隻讀更改為read-and-write。

1. a | array-like

輸入數組。

2. offset | int | optional

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

3. axis1 | int | optional

要從中提取對角線的第一個軸。默認情況下，軸1=0。

4. axis2 | int | optional

從中提取對角線的第二個軸。默認情況下，軸2=0。

僅當處理具有 3 個或更多維度的數組時，才需要考慮參數 axis1 和 axis2。

包含輸入數組對角線的 NumPy 數組。

考慮以下二維數組：

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



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

獲取主對角線：

np.diagonal(a)



array([1, 5, 9])

獲取偏移量為 1 的對角線：

np.diagonal(a, offset=1)



array([2, 6])

要獲得偏移量為 -2 的對角線：

np.diagonal(a, offset=-2)



array([7])

相關用法

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