matlab - 向量相乘时需要使用 if 语句：向量化

Question

我有一个矢量图

V=[ 1 2 3 5 9];

并想像这样乘以：

newVect=zeros(1,length(V));

for i=1:length(V)

    if V(i)==1
       newVect(i)=V(i)*somevaluex
    elseif V(i)==5
       newVect(i)=V(i)*somevaluey
    else
       newVect(i)=V(i);
end

这似乎很麻烦，因为我实际上正在处理具有 10000 个元素的数组。这可以以某种方式被矢量化取代吗？

Answer 1

你可以这样做：

newVect = V .* (   someValuex.*(V==1) +  someValuey.*(V==5) + V.*( (V~=1) & (V~=5) ) )

或者更快一点：

newVect=V;
newVect(V==1)=V(V==1)*someValuex;
newVect(V==5)=V(V==5)*someValuey;