如何在 Matlab 中裁剪任意维数的矩阵？

Question

假设我有 4D 矩阵：

>> A=1:(3*4*5*6);
>> A=reshape(A,3,4,5,6);

现在我想切割给定数量的行和列（或已知尺寸的任何给定块）。

如果我知道它是 4D 我会写：

>> A1=A(1:2,1:3,:,:);

但是如何针对任何给定数量的维度进行通用编写呢？

以下内容有所不同：

>> A2=A(1:2,1:3,:);

以下给出错误：

>> A2=A;
>> A2(3:3,4:4)=[];

Answer 1

可以使用您使用的第二种索引形式和 reshape 函数生成一般维数为A 的代码。 这里有一个例子：

Asize = [3,4,2,6,4]; %Initialization of A, not seen by the rest of the code
A = rand(Asize);

%% This part of the code can operate for any matrix A
I = 1:2;
J = 3:4;
A1 = A(I,J,:);
NewSize = size(A);
NewSize(1) = length(I);
NewSize(2) = length(J);
A2 = reshape(A1,NewSize);

A2 将是您裁剪的矩阵。它适用于您选择的任何Asize。

Answer 2

我推荐 Luis Mendo 针对一般情况提出的解决方案，但是当您知道尺寸的上限时，还有一个非常简单的解决方案。假设您最多有 6 个维度。对所有矩阵使用 6 维索引：

A1=A(1:2,1:3,:,:,:,:);

Matlab 将隐式假设所有剩余维度的单一维度，也为具有较少维度的矩阵返回预期结果。

Answer 3

听起来你只是想使用 ndims。

num_dimensions = ndims(A)

if (num_dimensions == 3)
    A1 = A(1:2, 1:3, :);
elseif (num_dimensions == 4)
    A1 = A(1:2, 1:3, :, :);
end

如果可能的矩阵维度范围很小，这种 if-else 块会保持简单。似乎您想要某种方法来动态创建索引元组（例如 (1:2,:,:,:) ），我不知道是否有办法。您必须将正确的维数与索引相匹配...如果您索引的维数少于矩阵的维数，matlab 会返回一个值，其中未索引的维数会折叠到一个数组中（类似于您得到的

A1 = A(:);