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=\left[\begin{array}{cccc}0.01& 0& 0& 0\\ 0& 0.01& 0& 0\\ 0& 0& 3.046\times 10^{-4}& 0\\ 0& 0& 0& 1\end{array}\right]$$

**

这是每个元素平方。

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