Python 将Numpy多维数组添加到一起

Python 将Numpy多维数组添加到一起,python,numpy,multidimensional-array,Python,Numpy,Multidimensional Array,如果我有两个Numpy数组:都是形状为(50,5,5)的数组,我如何将这两个数组相加得到一个形状为(50,5,10)的数组 import numpy In[2]: out_array = numpy.dstack([numpy.empty((50, 5, 5)), numpy.empty((50, 5, 5))]) In[3]: out_array.shape Out[3]: (50L, 5L, 10L) 使用: import numpy as np n = 50 * 5 * 5 a = np

如果我有两个Numpy数组:都是形状为(50,5,5)的数组,我如何将这两个数组相加得到一个形状为(50,5,10)的数组

import numpy
In[2]: out_array = numpy.dstack([numpy.empty((50, 5, 5)), numpy.empty((50, 5, 5))])
In[3]: out_array.shape
Out[3]: (50L, 5L, 10L)
使用:

import numpy as np
n = 50 * 5 * 5
a = np.random.random(size=n).reshape(50,5,5)
b = np.random.random(size=n).reshape(50,5,5)

np.concatenate([a,b], axis=2).shape # (50, 5, 10)