如何在 MATLAB 中使用 2-D 掩码索引 3-D 矩阵？

假设我有 D，一个 X×Y×Z 数据矩阵。我还有 M，一个 X×Y“掩蔽”矩阵。我的目标是当 M 中的 (Xi,Yi) 为 false 时，将 D 中的元素 (Xi,Yi,:) 设置为 NaN。

有什么办法可以避免在循环中这样做吗？我尝试使用ind2sub，但是失败了：

M = logical(round(rand(3,3))); % mask
D = randn(3,3,2); % data

% try getting x,y pairs of elements to be masked
[x,y] = ind2sub(size(M),find(M == 0));
D_masked = D;
D_masked(x,y,:) = NaN; % does not work!

% do it the old-fashioned way
D_masked = D;
for iX = 1:size(M,1)
    for iY = 1:size(M,2)
        if ~M(iX,iY), D_masked(iX,iY,:) = NaN; end
    end
end

我怀疑我在这里遗漏了一些明显的东西。 （：

您可以通过复制逻辑掩码来做到这一点M跨越三维空间使用REPMAT所以它的大小与D。然后，索引离开：

D_masked = D;
D_masked(repmat(~M,[1 1 size(D,3)])) = NaN;

如果不希望复制掩模矩阵，则还有另一种选择。您可以首先找到一组线性索引，其中M等于 0，然后复制该集合size(D,3)次，然后将每组索引移动多个numel(M)所以它索引了不同的部分D在三维空间中。我将在这里使用BSXFUN:

D_masked = D;
index = bsxfun(@plus,find(~M),(0:(size(D,3)-1)).*numel(M));
D_masked(index) = NaN;