带 if 语句的可向量化 FIND 函数 MATLAB

我有一个矩阵u，我想遍历所有行和所有列并执行以下操作。如果元素非零，我返回行索引的值。如果元素为零，则查找该元素之后的下一个非零元素的行索引。我可以使用两个带有 find 函数的 for 循环轻松完成此操作。但我需要多次执行此操作（不是因为矩阵的大小，而是因为这被调用了很多次）。我怎样才能做得更快？

这是for循环代码：

     for w=scenario_size:-1:1
            for t=1:time_size
                l = u(t,w) ;
                if l~=0
                    tprime = t ;
                else
                    tprime = t+ find(u(t:end,w),1,'first') -1 ;
                end
                i(t,w)       = tprime ;
                boo(t,w)     = number(tprime,w)/u(tprime,w) ;
            end
end

例如，如果一列是[0,0,5,1,0,3], the i is [3,3,3,4,6,6]。任意列的最后一个元素u始终非零（我通过在末尾人为地添加一行来强制这一点）。

Then boo是对应的条目tprime对于某些矩阵number除以相应的u（通过构造，其不为零）。

您可以使用以下方法解决此问题find https://www.mathworks.com/help/matlab/ref/find.html, cummin https://www.mathworks.com/help/matlab/ref/cummin.html， 还有一些逻辑索引 https://www.mathworks.com/help/matlab/math/array-indexing.html#MatrixIndexingExample-3。从这个例子开始：

>> u = randi([0 1], 10);
>> u(end, :) = 1

u =

     0     0     0     0     1     0     1     0     1     1
     1     0     1     1     1     0     0     0     0     1
     0     0     0     0     0     1     0     1     1     0
     1     1     1     0     1     1     0     0     0     0
     0     0     1     0     0     0     0     0     1     0
     1     1     1     0     0     0     0     0     0     0
     0     1     0     1     0     0     1     1     0     1
     1     1     0     1     0     1     1     1     1     1
     1     0     0     1     0     0     1     1     1     1
     1     1     1     1     1     1     1     1     1     1

以下将做你想做的事：

i = nan(size(u));  % Start with all nan values
[r, ~] = find(u);  % Get row indices of non-zero values
i(u ~= 0) = r;     % Place row indices in locations of non-zero values
i = cummin(i, 1, 'reverse');  % Column-wise cumulative minimum, starting from bottom

结果：

i =

     2     4     2     2     1     3     1     3     1     1
     2     4     2     2     2     3     7     3     3     2
     4     4     4     7     4     3     7     3     3     7
     4     4     4     7     4     4     7     7     5     7
     6     6     5     7    10     8     7     7     5     7
     6     6     6     7    10     8     7     7     8     7
     8     7    10     7    10     8     7     7     8     7
     8     8    10     8    10     8     8     8     8     8
     9    10    10     9    10    10     9     9     9     9
    10    10    10    10    10    10    10    10    10    10

然后你可以计算你的矩阵boo通过转换i to a 线性指数 https://www.mathworks.com/help/matlab/math/array-indexing.html#MatrixIndexingExample-2:

index = i+time_size.*repmat(0:(scenario_size-1), time_size, 1);  % Create linear index
boo = number(index)./u(index);

或者，您可以计算i作为从一开始的线性索引：

i = nan(size(u));  % Start with all nan values
index = find(u);   % Get linear indices of non-zero values
i(index) = index;  % Place linear indices in locations of non-zero values
i = cummin(i, 1, 'reverse');  % Column-wise cumulative minimum, starting from bottom
boo = number(i)./u(i);