如何从Matlab中保留其大小的矩阵中删除Inf值

Question

我在循环中有一个矩阵

m=


 -132.5901    -137.2695   -114.1264   -131.4986   -134.5733       Inf
       Inf       Inf       Inf        -135.2631   -121.7322       -119.7471
 -132.7978    -123.8068   -135.9385       Inf      -134.1602     -117.6997
 -130.1930    -134.0093   -137.4125   -128.7641       Inf        -116.0473

我想要一个可以在循环中使用的命令来得到答案:

 -132.5901 -137.2695 -114.1264 -131.4986 -134.5733  -119.7471  
 -132.7978 -123.8068 -135.9385 -135.2631 -121.7322  -117.6997
 -130.1930 -134.0093 -137.4125 -128.7641 -134.1602  -116.0473

我不想使用isfinite命令的单个向量.应使用Matlab自动保留矩阵大小.

Answer 1

我将假设在保留原始矩阵的大小时,您希望我们填充解决方案矩阵的底部NaN.我还假设Inf在任何给定的列中可能有多个.在这种情况下,以下基于循环的解决方案适用于我:

%# Set up an example matrix
M = [Inf 2 Inf 4; 5 Inf 6 7; Inf Inf 8 Inf];
[T, N] = size(M);

%# Get an index of finite elements
I1 = isfinite(M);

%# Solution 1
Soln1 = NaN(T, N);
for n = 1:N
    CurCol = M(:, n);
    Soln1(1:sum(I1(:, n)), n) = CurCol(I1(:, n));
end

对于一般情况,可能存在针对此问题的完全向量化解决方案(肯定有 - 请参阅下面的更新).但是,如果它比基于循环的解决方案提高了很多速度,我会感到惊讶.单圈现在在Matlab中非常快你如果你设置问题所以你可以操作矩阵的列(而不是行),因为这些元素在内存中顺序分配.

现在,让我们假设Inf每列只有一个.在这种情况下,问题更容易解决,可以通过以下单行完成:

%# Solution 2
Soln2 = [reshape(M(isfinite(M)), T-1, N); NaN(1, N)];

对于我设置的示例矩阵,解决方案2显然会失败,因为6个元素不能通过N(即2乘4)矩阵重新排列成T-1.

更新:好的,所以没有充分的理由,如果没有循环我就无法做到这一点令我烦恼.因此,这是一个无环路解决方案:

%# Solution 3
Soln3 = NaN(T, N);
I2 = bsxfun(@plus, cumsum(isfinite(M)), (0:T:T*(N-1)));
I2 = I2(:);
I2(I1 == 0) = [];
Soln3(I2) = M(isfinite(M));

对循环与非循环解决方案进行快速(非常)非严格的速度测试:

Elapsed time is 0.203997 seconds.
Elapsed time is 0.251969 seconds.

如果M更大,非循环解决方案可能会改善(相对而言).

现在我需要重新开始工作:-)