python - 如何获得 numpy 中对角线下的值总和？

Question

给定一个二维矩形 numpy 数组:

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

我想取左下角到右上角对角线下所有值的总和，即8、9 和 6。

实现此目标的最佳方法是什么？

该方法也适用于大型数组。

Answer 1

您可以使用 np.flip + np.tril + np.sum :

import numpy as np

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

print(np.sum(np.tril(np.flip(a, 1), -1)))
# 23