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ù)詳解
官方文檔
以一維數(shù)組的形式返回方陣的對(duì)角線(或非對(duì)角線)元素,或?qū)⒁痪S數(shù)組轉(zhuǎn)換成方陣(非對(duì)角線元素為0).兩種功能角色轉(zhuǎn)變?nèi)Q于輸入的v。1
更深層的見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è)參考,也希望大家多多支持腳本之家。
您可能感興趣的文章:- Python NumPy灰度圖像的壓縮原理講解
- Python多進(jìn)程共享numpy 數(shù)組的方法
- python中sqllite插入numpy數(shù)組到數(shù)據(jù)庫的實(shí)現(xiàn)方法
- python圖像處理基本操作總結(jié)(PIL庫、Matplotlib及Numpy)
- python numpy中multiply與*及matul 的區(qū)別說明
- 淺談Python numpy創(chuàng)建空數(shù)組的問題
- Python機(jī)器學(xué)習(xí)三大件之一numpy
- python利用numpy存取文件案例教程