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

Numpy 的 diagflat(~) 方法創建一個 2D Numpy 數組，其對角線由展平的輸入數組指定。所有其他條目均用零填充。

1. a | array-like

輸入數組。多維數組將自動展平為一維。

2. k | int | optional

如果 k 為正，則頂部的對角線將被填充。如果為負，則底部的對角線將被填充。默認情況下，k=0 。

二維 Numpy 數組。

考慮以下二維數組：

np.diagflat([4,5,6])



array([[4, 0, 0],
       [0, 5, 0],
       [0, 0, 6]])

np.diagflat([4,5,6], k=1)



array([[0, 4, 0, 0],
       [0, 0, 5, 0],
       [0, 0, 0, 6],
       [0, 0, 0, 0]])

np.diagflat([4,5], k=1)



array([[0, 4, 0],
       [0, 0, 5],
       [0, 0, 0]])

np.diagflat([4,5], k=-1)



array([[0, 0, 0],
       [4, 0, 0],
       [0, 5, 0]])

相關用法

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