如何生成對角矩陣 numpy.diag
更新時間:2022年05月11日 08:59:58 作者:yftadyz
這篇文章主要介紹了如何生成對角矩陣 numpy.diag,具有很好的參考價值,希望對大家有所幫助。如有錯誤或未考慮完全的地方,望不吝賜教
生成對角矩陣 numpy.diag
給定對角線上元素,我想生成對角矩陣,在網上搜了一下,竟然都是numpy.diagonal。
這個函數的作用是提取給定矩陣的對角元素,當然不是我想要的。
后來發(fā)現numpy.diag才是生成對角矩陣的函數,所以寫此文章記錄之。
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()結構及用法||參數
numpy.diag(v,k=0)
以一維數組的形式返回方陣的對角線(或非對角線)元素,或將一維數組轉換成方陣(非對角線元素為0).兩種功能角色轉變取決于輸入的v。1
更深層的見numpy.diagnal()
參數詳解
1.v:array_like.
如果v是2D數組,返回k位置的對角線。
如果v是1D數組,返回一個v作為k位置對角線的2維數組。
2.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]])
以上為個人經驗,希望能給大家一個參考,也希望大家多多支持腳本之家。
您可能感興趣的文章:
相關文章
python環(huán)境中的概念conda中與環(huán)境相關指令操作
這篇文章主要介紹了python環(huán)境中的概念conda中與環(huán)境相關指令操作,虛擬環(huán)境是從電腦獨立開辟出來的環(huán)境,文章介紹了相關概念,需要的朋友可以參考下2023-03-03