将矩阵拆分为具有不同列大小的数组

Question

我想将矩阵拆分为具有不同列大小的数组。我可以用 for 循环来完成它，但是我很好奇是否可以使用一些命令以更快的方式完成它。

举例来说，我有以下矩阵：

matrix = [[1, 2, 3, 4]
          [5, 6, 7, 8]
          [9, 10, 11, 12]
          [13, 14, 15, 16]]

现在我想获得一个二维数组，如下所示：

desired_array = [[1]
                 [5, 6]
                 [9, 10, 11]
                 [13, 14, 15, 16]]

我想要这个，因为我想要 desired_array 每行的总和。也许有另一种解决方案可以在不使用 for 循环的情况下获得该总和？

谢谢！

Answer 1

您只需要下三角矩阵的逐行总和。

>>> np.tril(matrix).sum(1)
array([ 1, 11, 30, 58])

Splitting a matrix into arrays with different column sizes