Python numpy.fill_diagonal()用法及代碼示例

借助numpy.fill_diagonal()方法，我們可以將numpy數組的對角線填充為參數中傳遞的值numpy.fill_diagonal()方法。

用法： numpy.fill_diagonal(array, value)

返回：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]]

相關用法

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