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]])

请注意在填充下一个对角线之前如何跳过单行。