NumPy包中的內(nèi)置diag函數(shù)很有意思。

假設(shè)創(chuàng)建一個(gè)1維數(shù)組a，和一個(gè)3*3數(shù)組b：

import numpy as np
a = np.arange(1, 4)
b = np.arange(1, 10).reshape(3, 3)

結(jié)果如下：

>>> a
array([1, 2, 3])
>>> b
array([[1, 2, 3],
[4, 5, 6],
[7, 8, 9]])

使用diag函數(shù)，看一看結(jié)果：

>>> np.diag(a)
array([[1, 0, 0],
[0, 2, 0],
[0, 0, 3]])
>>> np.diag(b)
array([1, 5, 9])

可以發(fā)現(xiàn)，當(dāng) np.diag(array) 中

array是一個(gè)1維數(shù)組時(shí)，結(jié)果形成一個(gè)以一維數(shù)組為對(duì)角線元素的矩陣

array是一個(gè)二維矩陣時(shí)，結(jié)果輸出矩陣的對(duì)角線元素

補(bǔ)充：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ù)詳解：

v : array_like.

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

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

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

以上為個(gè)人經(jīng)驗(yàn)，希望能給大家一個(gè)參考，也希望大家多多支持本站。

國(guó)外服務(wù)器租用

版權(quán)聲明：本站文章來(lái)源標(biāo)注為YINGSOO的內(nèi)容版權(quán)均為本站所有，歡迎引用、轉(zhuǎn)載，請(qǐng)保持原文完整并注明來(lái)源及原文鏈接。禁止復(fù)制或仿造本網(wǎng)站，禁止在非www.sddonglingsh.com所屬的服務(wù)器上建立鏡像，否則將依法追究法律責(zé)任。本站部分內(nèi)容來(lái)源于網(wǎng)友推薦、互聯(lián)網(wǎng)收集整理而來(lái)，僅供學(xué)習(xí)參考，不代表本站立場(chǎng)，如有內(nèi)容涉嫌侵權(quán)，請(qǐng)聯(lián)系alex-e#qq.com處理。