numpy ndarray 提取对角元素

方式1

import numpy as np
a=np.array([[1.123,2.123,3.123],[4.456,5.456,6.456],[7.789,8.789,9.789]])
print(a.diagonal(-1))#数字代表偏移量

结果：

array([4.456, 8.789])

方式2

使用库函数

np.diag(a,-1)#命令行

结果：

array([4.456, 8.789])