Python| numpy.fill_diagonal() 方法

借助numpy.fill_diagonal()方法，我们可以用numpy.fill_diagonal()方法中作为参数传递的值填充 numpy 数组的对角线。

Syntax : numpy.fill_diagonal(array, value)
Return : Return the filled value in the diagonal of an array.

示例 #1：
在这个例子中，我们可以看到，通过使用numpy.fill_diagonal()方法，我们能够用作为参数传递的值填充对角线。

# import numpy
import numpy as np
  
# using numpy.fill_diagonal() method
array = np.array([[1, 2], [2, 1]])
np.fill_diagonal(array, 5)
  
print(array)

输出 ：

[[5 2]
[2 5]]

示例 #2：

# import numpy
import numpy as np
  
# using numpy.fill_diagonal() method
array = np.zeros((3, 3), int)
np.fill_diagonal(array, 1)
  
print(array)

输出 ：

[[1 0 0]
[0 1 0]
[0 0 1]]