Python - 如何在不使用 numpy 的情况下获取矩阵的上三角之和？

Question

这个问题类似于this one但我想知道如何在不使用 numpy 的情况下做到这一点。我怎样才能用纯Python得到矩阵上三角的夏天？例如，我有

matrix = [[1,  2,  3],
          [4,  5,  6],
          [7,  8,  9]]

如何返回:

upper = [2,3,6]
upperSum = 11
lower = [4,7,8]
lowerSum = 19

Answer 1

对于方阵: 实际上，我认为即使对于非方阵，它的行为也是正确的。

>>> m
[[1, 2, 3], [4, 5, 6], [7, 8, 9]]
>>> sum(( m[i][i+1:] for i in range(len(m)) ), [])
[2, 3, 6]
>>> sum(( m[i][:i] for i in range(len(m)) ), [])
[4, 7, 8]

(使用sum-flatten hack)

>>> sum([[1, 2], [3, 4]], [])
[1, 2, 3, 4]