如何将Python中的三维数字数组乘成二维数字数组？
如何将Python中的三维数字数组乘成二维数字数组？
提问于 2020-11-07 10:47:26
假设我有以下数组A，其中右边是它的‘.side’：
A.shape = (10000, 10, 10)
我想得到这样的C.shape = (10000,10)：对于A中的每一个10x10矩阵(在10000个矩阵中)，它被简化为每一行之和的一维向量(10,1)，因此，最终的结果是C.shape = (10000,10)。
本质上，如果我们有形状(10,10)并乘以一个numpy.ones向量(10,1)，就可以完成这项工作。
但是在处理3D数组(10000,10,10)时，如何用Python编写这个呢？
最终目标是形状(10000,10)乘以= (10000,10)，其中(10000，10)的第二维现在是前(10,10)矩阵的行和。
回答 2
Stack Overflow用户
发布于 2020-11-07 10:59:52
是的，您可以使用.sum()沿轴1从(10000,10,10)形状数组获得它：
import numpy as np
a=np.ones(1000000).reshape(10000,10,10)
print(a.shape) # 10000,10,10
b=a.sum(axis=1)
print(b.shape) # 10000,10
c=b.sum(axis=1)
print(c.shape) # 10000,1Stack Overflow用户
发布于 2020-11-07 12:56:15
最简单的解决办法是：
import numpy as np
C = np.sum (A, axis=2)替代解决方案1(减少)：
如果要使用reduce()函数，请使用以下命令：
import numpy as np
C = np.add.reduce (A, 2)在这里，2表示通过add操作进行还原的轴。
有关详细信息，请参阅这
这可能只适用于1.17版及以上版本的numpy
替代方案2：(与1向量相乘)
使用numpy.dot()
B = np.ones((10,), dtype=A.dtype)
C = np.dot (A,B)页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接：
https://stackoverflow.com/questions/64726866
复制相关文章
相似问题
腾讯云开发者
Copyright © 2013 - 2026 Tencent Cloud. All Rights Reserved. 腾讯云 版权所有
深圳市腾讯计算机系统有限公司 ICP备案/许可证号：粤B2-20090059 
粤公网安备44030502008569号
腾讯云计算（北京）有限责任公司 京ICP证150476号 | 京ICP备11018762号