Python 结合了两个 ndarray

Question

组合两个 numpy 数组的最 pythonic 方法是什么，这样每个数组

(x,y,z) and (x1,y2,z3)

合并为

(2,x,y,z)

两个列表堆叠在哪里？是np.stack吗？

编辑：澄清一下，两个数组仍然存在，因此，

np.array([x,y,z],
         [x1,y2,z2])

因此，当索引到第一个元素时，您将获得第一个元组。第二个元素获取第二个元组。

Answer 1

是的，只需使用np.stack，它就可以随心所欲地工作！例如：

x = np.arange(100).reshape(5, 2, 10) 
x.shape # 5x2x10
np.stack((x, x)).shape # become 2x5x2x10

它沿前面创建的新轴连接其输入。但是，如果您有很多想要连接的数组，我建议您将它们转换为列表，然后进行连接并再次将结果转换为 numpy 数组，它会更快。