MATLAB 从矩阵中删除 NaN 值并将值左移

Question

我正在尝试计算以下矩阵中的列差异:

A = 
     0       NaN       NaN    0.3750       NaN
     NaN    0.1250    0.2500  0.3750       NaN

我想获得:

0.3750 NaN    NaN
0.1250 0.1250 0.1250

我本质上是在进行列差异，跳过 NaN 值并将值向左移动。

一维情况是:

A = [0 NaN 0.250 0.375 NaN 0.625];
NaN_diff(A) = [0.250 0.125 0.250];

有什么方法可以在 MATLAB 中高效地完成此操作，而无需每行使用低效的 find() 查询？

Answer 1

这是一个将大部分操作矢量化的解决方案:

notNan = ~isnan(A);
numNN  = sum(notNan,2);

shifted = NaN(size(A));

for r = 1:size(A,1)
   myRow = A(r,:);
   shifted(r,1:numNN(r)) = myRow(notNan(r,:));
end

nanDiff = diff(shifted,1,2);