在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