numpy.diag

numpy.diag(v, k=0)

提取對角線或建構對角線陣列。

如果您使用此函數提取對角線並希望寫入結果陣列，請參閱 numpy.diagonal 的更詳細文件；它返回副本還是視圖取決於您使用的 numpy 版本。

參數:

v類陣列

如果 v 是 2 維陣列，則傳回其第 k 條對角線的副本。如果 v 是 1 維陣列，則傳回一個 2 維陣列，其第 k 條對角線上為 v。

k整數，選填

有問題的對角線。預設值為 0。使用 k>0 表示主對角線以上的對角線，而 k<0 表示主對角線以下的對角線。

回傳值:

outndarray

提取的對角線或建構的對角線陣列。

參見

diagonal

回傳指定的對角線。

diagflat

建立一個 2 維陣列，其對角線為展平的輸入。

trace

沿對角線求和。

triu

陣列的上三角形。

tril

陣列的下三角形。

範例

>>> import numpy as np
>>> x = np.arange(9).reshape((3,3))
>>> x
array([[0, 1, 2],
       [3, 4, 5],
       [6, 7, 8]])

>>> np.diag(x)
array([0, 4, 8])
>>> np.diag(x, k=1)
array([1, 5])
>>> np.diag(x, k=-1)
array([3, 7])

>>> np.diag(np.diag(x))
array([[0, 0, 0],
       [0, 4, 0],
       [0, 0, 8]])