时间戳

首页 > 解决方案 > 将矩阵拆分为具有不同列大小的数组

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

问题描述

我想将矩阵拆分为具有不同列大小的数组。我可以使用 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]]

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

谢谢！

标签： arrayspython-3.xnumpyslicereshape

解决方案

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

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

推荐阅读