Python：如何將多個陣列堆疊在一起沿新軸

Question

我有三個矩陣，R，B，G，它們的大小都相同，比如說m * n。我想他們三個都疊加在一起成爲一個新的矩陣（3 * M * N或M * N * 3），如在Matlab中實現的不同的信道：

A(:,:,1) = R 
A(:,:,2) = G 
A(:,:,3) = B 

如何有效地實現這一目標蟒蛇？

Answer 1

numpy.stack應爲三個數組工作：

numpy.stack((R, G, B)) 

例如，

a = numpy.array([[1,2],[2,1]]) 
b = numpy.array([[3,3],[4,4]]) 
c = numpy.array([[5,6],[7,8]]) 
print(numpy.stack((a, b, c))) 

打印

[[[1 2] 
    [2 1]] 

[[3 3] 
    [4 4]] 

[[5 6] 
    [7 8]]]