arrays - 如何找到每行中非零元素的位置？

Question

给定一个矩阵A是

A=[  0     1     1
     1     0     1]

如何在不使用循环的情况下找到 A 矩阵每一行中非零值的位置。预期的结果喜欢

output=[2 3
        1 3]

我使用了find函数，但它返回了意想不到的结果

 output=[2
         3
         5
         6]

Answer 1

方法 #1

使用find获取展平数组中的列索引，然后 reshape -

[c,~] = find(A.')
out = reshape(c,[],size(A,1)).'

sample 运行-

>> A
A =
     0     1     1
     1     0     1
     1     1     0
>> [c,~] = find(A.');
>> reshape(c,[],size(A,1)).'
ans =
     2     3
     1     3
     1     2

---

方法 #2

我们可以通过一些排序避免输入数组的转置 -

[r,c]  = find(A);
[~,idx] = sort(r);
out = reshape(c(idx),[],size(A,1)).'

---

基准测试

我们将平铺行以形成更大的输入矩阵并测试所提出的方法。

基准代码-

% Setup input array
A0 = [ 0 1 1;1 0 1;1,1,0;1,0,1];
N = 10000000; % number of times to tile the input rows to create bigger one
A = A0(randi(size(A0,1),N,1),:);

disp('----------------------------------- App#1')
tic,
[c,~] = find(A.');
out = reshape(c,[],size(A,1)).';
toc
clear c out

disp('----------------------------------- App#2')
tic,
[r,c]  = find(A);
[~,idx] = sort(r);
out = reshape(c(idx),[],size(A,1)).';
toc
clear r c idx out

disp('----------------------------------- Wolfie soln')
tic,
[row, col] = find(A);
[~, idx] = sort(row);
out = [col(idx(1:2:end)), col(idx(2:2:end))];
toc

时间 -

----------------------------------- App#1
Elapsed time is 0.273673 seconds.
----------------------------------- App#2
Elapsed time is 0.973667 seconds.
----------------------------------- Wolfie soln
Elapsed time is 0.979726 seconds.

很难在 App#2 和@Wolfie 的解决方案之间做出选择，因为时间似乎相当，但第一个似乎非常有效。