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

例子

基本用法

考慮以下二維數組：

a = np.array([[5,6],[7,8]])
a



array([[5, 6],
       [7, 8]])

用值 2 填充對角線：

np.fill_diagonal(a, 2)
a



array([[2, 6],
       [7, 2]])

高矩陣

考慮以下高二維數組：

a = np.array([[1,1],[1,1],[1,1],[1,1],[1,1]])
a



array([[1, 1],
       [1, 1],
       [1, 1],
       [1, 1],
       [1, 1]])

不帶包裝

默認情況下，沒有環繞行為，因此隻會填充主對角線：

np.fill_diagonal(a, 0)
a



array([[0, 1],
       [1, 0],
       [1, 1],
       [1, 1],
       [1, 1]])

帶包裝

要啟用包裝行為，請像這樣設置wrap=True：

np.fill_diagonal(a, 0, wrap=True)
a



array([[0, 1],
       [1, 0],
       [1, 1],
       [0, 1],
       [1, 0]])

請注意在填充下一個對角線之前如何跳過單行。