将Matalab 60000x10数组转换为60000x1或1x60000

Question

使用大小为60000行，10列的2D数组数组如下

 [0 0 0 0 0 1 0 0 0 0]
 [1 0 0 0 0 0 0 0 0 0]
 [0 0 0 0 0 0 0 0 0 1]
.......

任何行将只包含一个'1'

我必须把它化简成一个行向量或者列向量来表示我们有一个1的索引。例如，对于上面显示的行，我们必须以

结尾

[6,1,10...]直到第60000个值。

我如何在没有循环的Matlab中做到这一点?

Answer 1

更新:更有效的方法是使用矩阵乘法:

idx = a * (1:size(a,2)).'

之前回答:

试试这个:

 a=[...
 0 0 0 0 0 1 0 0 0 0
 1 0 0 0 0 0 0 0 0 0
 0 0 0 0 0 0 0 0 0 1]
 [r ,~]=find(a.')

使用Divakar的基准测试:

------------------- With FIND
Elapsed time is 1.55641 seconds.
------------------- With MAX
Elapsed time is 0.703054 seconds.
------------------- With MTIMES
Elapsed time is 0.266676 seconds.

Answer 2

您可以使用max -

[~,idx] = max(a,[],2)

60000 x 10阵列运行测试

基准代码-

% Setup input array
a = zeros(60000,10);
idx = randi(size(a,2),1,size(a,1));
a(sub2ind(size(a),1:size(a,1),idx)) = 1;
% Timing code
disp('------------------- With FIND')
tic,for i = 1:500,[r,~]=find(a.');end, toc
disp('------------------- With MAX')
tic,for i = 1:500,[~,idx] = max(a,[],2);end, toc

计时,

------------------- With FIND
Elapsed time is 4.267364 seconds.
------------------- With MAX
Elapsed time is 1.353760 seconds.