介绍
diag(diagnol)是对角的意思。numpy.diag是一个用来提取矩阵对角数或创建对角阵的函数。
例子
调用
import numpy as np
construct
Q = np.diag([
0.1,
0.1,
np.deg2rad(1.0),
1.0
]) ** 2
就创建了下面这个矩阵:
Q=⎣⎢⎢⎡0.0100000.0100003.046×10−400001⎦⎥⎥⎤
**
这是每个元素平方。
extract
就以上面创建的矩阵为待提取矩阵。
np.diag(Q)
返回
array([0.01,0.01,3.046×10−4,1])
参考资料
https://docs.scipy.org/doc/numpy/reference/generated/numpy.diag.html