在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.]