在numpy中组合矩阵

Question

假设我有三个矩阵a，b和c的“片”，每个矩阵都有相同的p维。 （i，j，k），b [i，j，k]，c [i（i，j，k）]是一个新的矩阵，并且我想将它们组合起来得到一个新的矩阵。 ，J，K]）。我应该使用哪个命令？ dstack命令在这里似乎不起作用。谢谢。

Answer 1

我想你想要的是：

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

Answer 2

另外一个内胆是：

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