将Numpy Multi维数组加在一起

Question

如果我有两个Numpy数组：两个都有形状（50,5,5）我如何添加它们以获得一个带有（50,5,10）的数组？

Answer 1

使用concatenate：

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)

Answer 2

numpy.dstack可以解决问题

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)