arrays - 使用逻辑数组索引到矩阵

Question

我有一个矩阵A，它是m x n。我想要做的是计算一行中 NaN 元素的数量。如果 NaN 元素的数量大于或等于某个任意阈值，则该行中的所有值都将设置为 NaN。

num_obs = sum(isnan(rets), 2);
index = num_obs >= min_obs;

就像我说的，我正在努力让我的大脑工作。正在尝试下面这条线的不同变体，但没有成功。

rets(index==0, :) = rets(index==0, :) .* NaN;

threshold >= 1 的示例数据是:

A = [-7 -8 1.6 11.9;
   NaN NaN NaN NaN;
   5.5 6.3 2.1 NaN;
   5.5 4.2 2.2 5.6;
   NaN NaN NaN NaN];

我想要的结果是:

A = [-7 -8 1.6 11.9;
   NaN NaN NaN NaN;
   NaN NaN NaN NaN;
   5.5 4.2 2.2 5.6;
   NaN NaN NaN NaN];

Answer 1

使用

A = magic(4);A(3,3)=nan;
threshold=1;

for ii = 1:size(A,1) % loop over rows
    if sum(isnan(A(ii,:)))>=threshold % get the nans, sum the occurances
        A(ii,:)=nan(1,size(A,2)); % fill the row with column width amount of nans
    end
end

结果在

A =

    16     2     3    13
     5    11    10     8
   NaN   NaN   NaN   NaN
     4    14    15     1

或者，作为 @Obchardon在他的评论中提到你可以向量化:

A(sum(isnan(A),2)>=threshold,:) = NaN

A =

    16     2     3    13
     5    11    10     8
   NaN   NaN   NaN   NaN
     4    14    15     1

作为旁注，您可以轻松地将其更改为列，只需对其他维度进行所有索引即可:

A(:,sum(isnan(A),1)>=threshold) = NaN;