Matlab:使用给定条件更改 3D 数组中的元素

Matlab: Change elements in 3D array using given conditions

我在 Matlab 中有一个 3 维数组 (10x3x3),我想将任何大于 999 的值更改为 Inf。但是,我只想将其应用于此数组的 (:,:,2:3)。

我在网上找到的所有帮助似乎只适用于整个数组,或二维数组的 1 列。我不知道如何将其应用于 3D 阵列。

我已经尝试了下面的代码,但是在我 运行 它之后它变成了一个 69x3x3 的数组,我真的不明白为什么。我试图从使用二维数组的人那里复制代码,所以我只是觉得我不太明白代码在做什么。

A(A(:,:,2)>999,2)=Inf;
A(A(:,:,3)>999,3)=Inf;

一种方法 logical indexing -

mask = A>999;  %// get the 3D mask
mask(:,:,1) = 0; %// set all elements in the first 3D slice to zeros, 
%// to neglect their effect when we mask the entire input array with it
A(mask) = Inf %// finally mask and set them to Infs

另一个 linear indexing -

idx = find(A>999); %// Find linear indices that match the criteria
valid_idx = idx(idx>size(A,1)*size(A,2)) %// Select indices from 2nd 3D slice onwards
A(valid_idx)=Inf %// Set to Infs

或者另一个 linear indexing,几乎与前一个相同,有效索引是在一个步骤中计算的,因此使我们能够使用一行 -

A(find(A(:,:,2:3)>999) + size(A,1)*size(A,2))=Inf