怎么生成對(duì)角矩陣numpy.diag

本文小編為大家詳細(xì)介紹“怎么生成對(duì)角矩陣numpy.diag”，內(nèi)容詳細(xì)，步驟清晰，細(xì)節(jié)處理妥當(dāng)，希望這篇“怎么生成對(duì)角矩陣numpy.diag”文章能幫助大家解決疑惑，下面跟著小編的思路慢慢深入，一起來(lái)學(xué)習(xí)新知識(shí)吧。

生成對(duì)角矩陣 numpy.diag

給定對(duì)角線上元素，我想生成對(duì)角矩陣，在網(wǎng)上搜了一下，竟然都是numpy.diagonal。

這個(gè)函數(shù)的作用是提取給定矩陣的對(duì)角元素，當(dāng)然不是我想要的。

后來(lái)發(fā)現(xiàn)numpy.diag才是生成對(duì)角矩陣的函數(shù)，所以寫(xiě)此文章記錄之。

import numpy as np
a=[1,2,3]
np.diag(a)
Out[4]: 
array([[1, 0, 0],
       [0, 2, 0],
       [0, 0, 3]])

numpy.diag()結(jié)構(gòu)及用法||參數(shù)

numpy.diag(v,k=0)

官方文檔

以一維數(shù)組的形式返回方陣的對(duì)角線（或非對(duì)角線）元素，或?qū)⒁痪S數(shù)組轉(zhuǎn)換成方陣（非對(duì)角線元素為0）.兩種功能角色轉(zhuǎn)變?nèi)Q于輸入的v。1

更深層的見(jiàn)numpy.diagnal()

參數(shù)詳解

1.v：array_like.

如果v是2D數(shù)組，返回k位置的對(duì)角線。

如果v是1D數(shù)組，返回一個(gè)v作為k位置對(duì)角線的2維數(shù)組。

2.k：int, optional

對(duì)角線的位置，大于零位于對(duì)角線上面，小于零則在下面。

示例

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

讀到這里，這篇“怎么生成對(duì)角矩陣numpy.diag”文章已經(jīng)介紹完畢，想要掌握這篇文章的知識(shí)點(diǎn)還需要大家自己動(dòng)手實(shí)踐使用過(guò)才能領(lǐng)會(huì)，如果想了解更多相關(guān)內(nèi)容的文章，歡迎關(guān)注億速云行業(yè)資訊頻道。