matlab从函数中的3D矩阵中提取特定的2D矩阵[重复]

Question

我想从 3D 矩阵中提取 2D 矩阵。现在，我知道如何使用

A = ones(10,10,10);
B = squeeze(A(:,:,2));

但我想编写一个函数，它将 3D 矩阵作为输入，以及获取 2D 矩阵的维度和索引。在上面的示例中，这将给出：

B = my_func(A,3,2);

有什么想法吗？

非常感谢！

Answer 1

function out=my_func(A,dim,ix)
    index=repmat({':'},1,ndims(A));
    index{dim}=ix;
    out=squeeze(getfield(A,index));
end

或者另一种选择：

function out=my_func(A,dim,ix)
    index=arrayfun(@(x)(1:x),size(A),'uni',false);
    index{dim}=ix;
    out=squeeze(A(index{:}));
end

Answer 2

怎么样：

function out=my_func(A,dim,i)

switch dim
    case 1
        out = squeeze(A(i,:,:));
    case 2
        out = squeeze(A(:,i,:));
    case 3
        out = squeeze(A(:,:,i));
end