两条对角线之间的矩阵元素

假设我们有一个矩阵A，如何找到两条对角线（从左上角到右下角和从右上角到左下角之间的对角线）之间的所有元素？ 可以使用numpy库中的triu()和tril()函数来实现。

代码示例：

import numpy as np

# 创建一个矩阵A
A = np.array([[1,2,3],
              [4,5,6],
              [7,8,9]])

# 找到左上到右下的对角线之间的元素
upper_diag = np.triu(A, k=1).flatten()
print("左上到右下的对角线之间的元素:", upper_diag)

# 找到右上到左下的对角线之间的元素
lower_diag = np.tril(A, k=-1).flatten()
print("右上到左下的对角线之间的元素:", lower_diag)

运行结果：

左上到右下的对角线之间的元素: [2 3 6] 右上到左下的对角线之间的元素: [4 7 8]