Python NumPy中diag函數(shù)的使用說明
NumPy包中的內(nèi)置diag函數(shù)很有意思。
假設(shè)創(chuàng)建一個1維數(shù)組a，和一個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是一個1維數(shù)組時，結(jié)果形成一個以一維數(shù)組為對角線元素的矩陣
array是一個二維矩陣時，結(jié)果輸出矩陣的對角線元素
補(bǔ)充：numpy.diag()結(jié)構(gòu)及用法||參數(shù)詳解
numpy.diag(v,k=0)
以一維數(shù)組的形式返回方陣的對角線（或非對角線）元素，或?qū)⒁痪S數(shù)組轉(zhuǎn)換成方陣（非對角線元素為0）.兩種功能角色轉(zhuǎn)變?nèi)Q于輸入的v。1
更深層的見numpy.diagnal()
參數(shù)詳解：
v : array_like.
如果v是2D數(shù)組，返回k位置的對角線。
如果v是1D數(shù)組，返回一個v作為k位置對角線的2維數(shù)組。
k : int, optional
對角線的位置，大于零位于對角線上面，小于零則在下面。
示例
>>> 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]])
以上為個人經(jīng)驗，希望能給大家一個參考，也希望大家多多支持腳本之家。
相關(guān)文章
局域網(wǎng)內(nèi)python socket實現(xiàn)windows與linux間的消息傳送
這篇文章主要介紹了局域網(wǎng)內(nèi)python socket實現(xiàn)windows與linux間的消息傳送的相關(guān)知識，非常不錯，具有一定的參考借鑒價值,需要的朋友可以參考下2019-04-04
如果你之前沒用過進(jìn)度條，八成是覺得它會增加不必要的復(fù)雜性或者很難維護(hù)，其實不然。要加一個進(jìn)度條其實只需要幾行代碼，快跟隨小編一起學(xué)習(xí)學(xué)習(xí)吧2022-12-12
這篇文章主要介紹了Python實現(xiàn)猜年齡游戲代碼實例,文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值,需要的朋友可以參考下2020-03-03
隨著項目代碼行數(shù)的增加,不可避免的遇到軟件架構(gòu)腐敗的問題,所以如何寫出簡潔的代碼至關(guān)重要,這篇文章主要給大家介紹了一些讓Python代碼簡潔的實用技巧,需要的朋友可以參考下2021-08-08
Python標(biāo)準(zhǔn)庫之Math,Random模塊使用詳解
math數(shù)學(xué)模塊和random隨機(jī)模塊是Python常用的標(biāo)準(zhǔn)庫之一。本文將詳細(xì)為大家介紹一下這兩個模塊的使用方法，需要的小伙伴可以參考一下2022-05-05
Python操作MySQL數(shù)據(jù)庫的兩種方式實例分析【pymysql和pandas】
這篇文章主要介紹了Python操作MySQL數(shù)據(jù)庫的兩種方式,結(jié)合實例形式分析了Python使用pymysql和pandas模塊進(jìn)行mysql數(shù)據(jù)庫的連接、增刪改查等操作相關(guān)實現(xiàn)技巧,需要的朋友可以參考下2019-03-03
解決Python Matplotlib繪圖數(shù)據(jù)點(diǎn)位置錯亂問題
這篇文章主要介紹了解決Python Matplotlib繪圖數(shù)據(jù)點(diǎn)位置錯亂問題，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2020-05-05
很多網(wǎng)站登錄都需要輸入驗證碼,如果要實現(xiàn)自動登錄就不可避免的要識別驗證碼,這篇文章主要給大家介紹了關(guān)于python簡單驗證碼識別的實現(xiàn)過程,需要的朋友可以參考下2021-06-06
Python?OpenCV實現(xiàn)姿態(tài)識別的詳細(xì)代碼
這篇文章主要介紹了Python?OpenCV實現(xiàn)姿態(tài)識別的方法，本文通過截圖實例代碼相結(jié)合給大家介紹的非常詳細(xì)，對大家的學(xué)習(xí)或工作具有一定的參考借鑒價值,需要的朋友可以參考下2022-02-02
30分鐘搭建Python的Flask框架并在上面編寫第一個應(yīng)用
這篇文章主要介紹了如何搭建Python的Flask框架并在上面編寫一個簡單的登錄模版應(yīng)用,代碼數(shù)量少、充分體現(xiàn)了Flask框架的輕量與開發(fā)高效的特點(diǎn)，需要的朋友可以參考下2015-03-03