векторизовать фильтр для подпоследовательности массива в Matlab

Question

У меня есть вектор «a» и фильтр «b». Оба этих вектора содержат только 0 или 1. Я хотел бы преобразовать «a», так что любая последовательность из 1 начинается только тогда, когда b находится в 1.

Я проиллюстрировал это с помощью цикла, но, поскольку мои векторы огромны, он крайне неэффективен.

Результат, который я хотел бы, хранится в «r».

a=[0;0;1;1;1;1;1;1;0;0;1;1;0;0;1;1;1;1;1]; 

b=[0;0;0;0;1;0;1;0;0;1;0;1;0;1;1;0;0;0;0]; 

r=[0;0;0;0;1;1;1;1;0;0;0;1;0;0;1;1;1;1;1]; 

for i=2:length(a) 

    if a(i)==1 &&a(i-1)==0 && b(i)==0   
     a(i)=a(i-1);   
    end 
end 

assert(sum(a==r)==length(a)) 

Answer 1

Вот два лайнера:

r = a; 
r([false; diff(a)>0 & b(2:end)==0]) = 0; 

Пожалуйста, обратите внимание, что вам нужно адаптировать код для векторов-строк (это работает для векторов-столбцов).