在 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个矩阵(或page,见下图)的行和列;
有人知道如何在 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,1,2
#0.0
A[0,:,:] # 3x3 slice at index 0
#array([[ 0., 0., 0.],
# [ 0., 0., 0.],
# [ 0., 0., 0.]])
A[1,1,:] # 1-D array of length 3
#array([ 0., 0., 0.]
关于python - 如何在 python 中定义多维数组?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/41741319/