我有矩阵
A=[2 3 4 5 6 7;
7 6 5 4 3 2]
我想计算有多少元素的值大于 3 且小于 6。
flatA = A(:);
count = sum(flatA > 3 & flatA < 6);
我可以想到几个方法:
count = numel(A( A(:)>3 & A(:)<6 )) %# (1)
count = length(A( A(:)>3 & A(:)<6 )) %# (2)
count = nnz( A(:)>3 & A(:)<6 ) %# (3)
count = sum( A(:)>3 & A(:)<6 ) %# (4)
Ac = A(:);
count = numel(A( Ac>3 & Ac<6 )) %# (5,6,7,8)
%# prevents double expansion
%# similar for length(), nnz(), sum(),
%# in the same order as (1)-(4)
count = numel(A( abs(A-(6+3)/2)<3/2 )) %# (9,10,11,12)
%# prevents double comparison and &
%# similar for length(), nnz(), sum()
%# in the same order as (1)-(4)
所以,让我们测试一下最快的方法是什么。测试代码:
A = randi(10, 50);
tic
for ii = 1:1e5
%# method is inserted here
end
toc
结果(最好的 5 次运行,均以秒为单位):
%# ( 1): 2.981446
%# ( 2): 3.006602
%# ( 3): 3.077083
%# ( 4): 2.619057
%# ( 5): 3.011029
%# ( 6): 2.868021
%# ( 7): 3.149641
%# ( 8): 2.457988
%# ( 9): 1.675575
%# (10): 1.675384
%# (11): 2.442607
%# (12): 1.222510
所以这似乎count = sum(( abs(A(:)-(6+3)/2)<3/2 ));
是去这里的最佳方式。
就个人而言:我没想到在 Matlab 中比较会比算术慢——有人知道对此的解释吗?
nnz
另外:与 相比,为什么这么慢sum
?我想这是有道理的,因为我知道比较比算术慢......
length(A(A>3 & A<6))
Accumarray是用来做这些事情的:
count = accumarray(A(A>3 & A<6),1)
返回
>> count'
ans =
0 0 0 2 2
你可以总结:
count = sum(count);
第一个零对应于我们忽略的 1,2,3 的出现次数。
或者更简单:
count = sum(A(:)>3 & A(:)<6);
您可以使用 matlab for-loop 并自行循环遍历这些值。优点是可以指定任何函数(>2 & <5 ; >3 & <6; 等),缺点是它是一种繁重的方法。这是大概的代码:
count = 0;
for i=1:length(A)
element = A(i);
if (element > 2 && element < 5)
count = count + 1;
end
end