在MATLAB中,有一种简单的方法来定义多维数组,例如

A(:,:,1) = [1,2,3; 4,5,6];
A(:,2) = [7,8,9; 10,11,12];

>> A

 A(:,1) =

 1     2     3
 4     5     6


 A(:,2) =

 7     8     9
 10    11    12

其中前两个索引分别为存储在A中的第i个矩阵(或页面,见下图)的行和列;

有谁知道如何在python中定义相同的结构？

解决方法

与NumPy索引类似于MATLAB

import numpy as np
 A=np.empty((2,3,3))
 A.shape
 #(2L,3L,3L)
 A[0,1,2] # element at index 0,2
 #0.0
 A[0,:] # 3x3 slice at index 0
 #array([[ 0.,0.,0.],#       [ 0.,0.]])
 A[1,:] # 1-D array of length 3
 #array([ 0.,0.]