python生成單位陣或?qū)顷嚨娜N方式小結(jié)

更新時間：2022年05月11日 08:37:13 作者：小Aer

這篇文章主要介紹了python生成單位陣或?qū)顷嚨娜N方式小結(jié)，具有很好的參考價值，希望對大家有所幫助。如有錯誤或未考慮完全的地方，望不吝賜教

1.identity

np.identity(4)
array([[ 1., 0., 0., 0.],
       [ 0., 1., 0., 0.],
       [ 0., 0., 1., 0.],
       [ 0., 0., 0., 1.]])

2.eye

np.eye(4)
array([[1., 0., 0., 0.],
       [0., 1., 0., 0.],
       [0., 0., 1., 0.],
       [0., 0., 0., 1.]])

3.diag（可以指定對角元素）

np.diag([1] * 4)
Out[1]:?
array([[1, 0, 0, 0],
? ? ? ?[0, 1, 0, 0],
? ? ? ?[0, 0, 1, 0],
? ? ? ?[0, 0, 0, 1]])
np.diag([2] * 4)
Out[2]:?
array([[2, 0, 0, 0],
? ? ? ?[0, 2, 0, 0],
? ? ? ?[0, 0, 2, 0],
? ? ? ?[0, 0, 0, 2]])

有趣的地方是前兩種方法元素都是浮點(diǎn)數(shù)，最后一種是整數(shù)，使用的時候注意區(qū)分就好

python創(chuàng)建對角陣的np.eye()函數(shù)

最近博主在研究kalman濾波，里面初始矩陣定義需要對角陣，于是查了一些資料，發(fā)現(xiàn)numpy中有一個eye函數(shù)可以達(dá)到這樣的目的

np.eye(N,M=None,k=0,dtype=<class 'float'>,order='C)

N表示輸出的行數(shù)；
M表示輸出的列數(shù)，不給默認(rèn)等于N；
K默認(rèn)等于0，表示主對角線，負(fù)數(shù)代表低對角，正數(shù)代表高對角；
dtype表示輸出數(shù)據(jù)的類型；
order表示輸出的數(shù)組的形式是按照C語言的行優(yōu)先’C’，還是按照Fortran形式的列優(yōu)先‘F’存儲在內(nèi)存中。

看下面兩個例子就明白用法啦

    a = np.eye(4)
    print(type(a))
    print(a)
    a = np.mat(a)
    print(type(a))
    print(a)
    a = a.I
    print(type(a))
    print(a)
>>><class 'numpy.ndarray'>
[[1. 0. 0. 0.]
 [0. 1. 0. 0.]
 [0. 0. 1. 0.]
 [0. 0. 0. 1.]]
<class 'numpy.matrix'>
[[1. 0. 0. 0.]
 [0. 1. 0. 0.]
 [0. 0. 1. 0.]
 [0. 0. 0. 1.]]
<class 'numpy.matrix'>
[[1. 0. 0. 0.]
 [0. 1. 0. 0.]
 [0. 0. 1. 0.]
 [0. 0. 0. 1.]]

    a = np.eye(4,k=1)
    print(type(a))
    print(a)
    a = np.mat(a)
    print(type(a))
    print(a)
    a = a.T
    print(type(a))
    print(a)
>>><class 'numpy.ndarray'>
[[0. 1. 0. 0.]
 [0. 0. 1. 0.]
 [0. 0. 0. 1.]
 [0. 0. 0. 0.]]
<class 'numpy.matrix'>
[[0. 1. 0. 0.]
 [0. 0. 1. 0.]
 [0. 0. 0. 1.]
 [0. 0. 0. 0.]]
<class 'numpy.matrix'>
[[0. 0. 0. 0.]
 [1. 0. 0. 0.]
 [0. 1. 0. 0.]
 [0. 0. 1. 0.]]

動動小手，感悟一下就好啦！以上為個人經(jīng)驗(yàn)，希望能給大家一個參考，也希望大家多多支持腳本之家。

您可能感興趣的文章: