Python numpy diagflat用法及代碼示例

本文簡要介紹 python 語言中 numpy.diagflat 的用法。

用法:

numpy.diagflat(v, k=0)

創建一個二維數組，將扁平化的輸入作為對角線。

參數：

v： array_like

輸入數據，被展平並設置為輸出的k-th 對角線。

k： 整數，可選

對角線設置； 0，默認值，對應於 “main” 對角線，正(負)k 給出主上(下)對角線的編號。

返回：

out： ndarray

二維輸出數組。

例子：

>>> 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]])