在numpy中组合矩阵

Question

假设我有三张矩阵a，b和c的“张”，每张都有相同的m n p维度。我想将它们组合起来得到一个新的m n p * 3矩阵，其（i，j，k）元素是（a [i，j，k]，b [i，j，k ]，C [I，J，K]）。我应该使用哪个命令？ dstack命令似乎不起作用。感谢。

Answer 1

另一个班轮将是：

result = numpy.array( (a,b,c) ).transpose( (1,2,3,0) )

或更自我描述的方法：

result = empty( (m,n,p,3) )
result[:,:,:,0] = a
result[:,:,:,1] = b
result[:,:,:,2] = c

Answer 2

我认为你想要的是：

np.concatenate([np.expand_dims(x, -1) for x in (a, b, c)], axis=3)