vectorisé si dans matlab

Question

J'ai un tableau booléen l'appeler flag.

J'ai deux tableaux numériques ifTrue, ifFalse. Tous ces tableaux sont de la même taille. Pour les besoins de cette question, supposons que chaque élément de ces tableaux est unique.

Je voudrais une fonction g avec la propriété que

a = g(flag, ifTrue, ifFalse) 

all(flag == (a == ifTrue)) 
all(~flag == (a == ifFalse)) 

Ou en anglais, je voudrais revenir gifTrue éléments lorsque flag est vrai, et ifFalse éléments quand flag est faux.

Ou, dans Matlab, je pouvais le faire avec des boucles:

a = zeros(size(ifTrue)); 
for i = 1 : numel(ifTrue); 
    if flag(i) 
     a(i) = ifTrue(i) 
    else 
     a(i) = ifFalse(i) 
    end 
end 

Y at-il une approche vectorisé?

Merci

Answer 1

%# Given, for example: 
ifTrue = 1:10 
ifFalse = -ifTrue 
flag = rand(1,10) > 0.5 
%# First, set 'a' to ifFalse 
a = ifFalse 
%# Then override the places where flag is true 
a(flag) = ifTrue(flag) 

Answer 2

Supposant que l'indicateur contient les pour vrai, et des zéros pour les faux éléments: a = flag .* ifTrue + (1 - flag) .* ifFalse;