numpy.diagflat#

numpy.diagflat(v, k=0)[原始碼]#

建立一個二維陣列，其對角線為扁平化的輸入。

參數:

varray_like: 輸入資料，將被扁平化並設定為輸出的第 k 條對角線。
kint，選用: 要設定的對角線；0 為預設值，對應於「主」對角線，正（負）k 值表示主對角線之上（之下）的第幾條對角線。

回傳值:

outndarray: 2-D 輸出陣列。

另請參閱

diag: 用於 1-D 和 2-D 陣列的 MATLAB 相似功能。
diagonal: 回傳指定的對角線。
trace: 沿對角線求和。

範例

>>> import numpy as np
>>> np.diagflat([[1,2], [3,4]])
array([[1, 0, 0, 0],
       [0, 2, 0, 0],
       [0, 0, 3, 0],
       [0, 0, 0, 4]])

>>> np.diagflat([1,2], 1)
array([[0, 1, 0],
       [0, 0, 2],
       [0, 0, 0]])