將各種2D矩陣保存到python中的3D矩陣中

Question

我試圖將各種2D矩陣從python保存到3D矩陣中，但我收到一個錯誤，這里是矩陣：

rho=np.zeros((Nx,Nt))
p=np.zeros((Nx,Nt))
u=np.zeros((Nx,Nt))
U=np.zeros((Nx,Nt,3))
F=np.zeros((Nx,Nt,3))

我試圖將前三個的組合保存到最后兩個中，如下所示：

F[:][:][0]=u*rho
F[:][:][1]=p*u**2+p
F[:][:][2]=u(e + p)

知道e是另一部分代碼中的另一個NX by Nt矩陣，似乎我沒有正確分配這個，因為我得到一個錯誤，說我無法將1000 * 1000矩陣保存到1000 * 3矩陣中，這是可以理解，但當然我的想法是將1000 * 1000矩陣保存到我的3d矩陣F的第一個“元素”中，將其他組合保存到我的U矩陣等等，任何人都可以告訴我如何做到這一點，保存2d矩陣進入3d矩陣的索引，我也試過U[0][:][:]=rho

Answer 1

正如Divakar指出的那樣，我沒有正確使用索引，使用F[:,:,0]=u*rho等，糾正了問題。