Question

我的代码的简化版本如下：

A = linspace(0,10,100);
threshold = 6.0;

我希望将矢量A截断为低于阈值的值。

假设A中的值总是在升序，我怎样才能整齐有效地做到这一点？

目前我能想到的唯一方法是引入for循环并逐个检查每个元素，如果它的值高于阈值，然后将此值分配给新数组。像这样：

    k=1;
    for i = 1:numel(A)
        if A(i) < threshold
        elseif A(i) >= threshold
            Atrunc(k,1) = A(i);
            k=k+1;
        end
    end

然而，这看起来并不是很好。对我来说，任何人都可以提供更优化的代码......？

Answer 1

使用逻辑索引

A = A(A < threshold);

或

A = A(A >= threshold);

Answer 2

由于A中的值是升序，您可以使用binary search查找低于阈值的最后一个元素：

threshold = 5; %// example data
A = linspace(0,15,1e7); %// example data

w = [1 numel(A)]; %// uncertainty window. Initiallize
while w(2)>w(1)+1
    t = round((w(1)+w(2))/2); %// test middle point
    if A(t)<threshold
        w(1) = t; %// remove lower half of window
    else
        w(2) = t; %// remove upper half of window
    end
end
if A(w(2)) < threshold %// handle special cases: Atrunc is A or is []
    t = w(2);
elseif A(w(1)) < threshold 
    t = w(1);
else
    t = w(1)-1;
end
Atrunc = A(1:t);

对于A大的这种方法（利用A已排序的事实）可能比Atrunc = A(A < threshold);更快

截断数组

2 个答案: