【问】numpy二维数组得对角线与反对角线

1. numpy二维数组得对角线

import numpy as np
a=np.array([[3,1,5],
           [6,7,9],
           [8,2,4]])
print(a)

• 正对角线

b=np.diag(a) # 主对角线
b1=np.diag(a,k=1) # 上次对角线
b=np.diag(a,k=1) #下次对角线

out:
[3,7,4]
[1,9]
[6,2]

• 反对角线

step1:左右翻转
np.fliplr()
矩阵左右翻转，在二维的情况下很容易理解
setp2:求对角线

c=np.fliplr(a) # 左右翻转
cdiag=np.diag(c)