Python numpy.fill_diagonal()方法

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

语法: numpy.fill_diagonal(array, value)
返回：返回数组对角线上的填充值。

例子#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]]