孤男寡女在一個向量乘以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;