如何解决将矩阵拆分为具有不同列大小的数组
我想将矩阵拆分为具有不同列大小的数组。我可以使用 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])
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。