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

问题描述

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

例如，假设我有以下矩阵：

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

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

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

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

谢谢！

解决方法

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

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