numpy.diag怎么用

介绍

diag(diagnol)是对角的意思。numpy.diag是一个用来提取矩阵对角数或创建对角阵的函数。

例子

调用

import numpy as np

construct

Q = np.diag([
    0.1,  
    0.1,  
    np.deg2rad(1.0), 
    1.0 
]) ** 2  

就创建了下面这个矩阵：
$Q=\begin{bmatrix} 0.01 & 0 & 0 & 0\\ 0 & 0.01 & 0 & 0\\ 0 & 0 & 3.046\times 10^{-4} & 0\\ 0 & 0 & 0 & 1 \\ \end{bmatrix}$

**

这是每个元素平方。

extract

就以上面创建的矩阵为待提取矩阵。

np.diag(Q)

返回 $array([0.01,0.01,3.046\times 10^{-4} ,1])$

参考资料

https://docs.scipy.org/doc/numpy/reference/generated/numpy.diag.html