将各种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等，纠正了问题。