如果在matlab中矢量化

Question

我有一个布尔数组调用它flag.

我有两个数字数组ifTrue,ifFalse.所有这些数组都是相同的大小.出于这个问题的目的,假设这些数组中的每个元素都是唯一的.

我想要一个g具有该属性的函数

a = g(flag, ifTrue, ifFalse)

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

或者用英语,我想g在if 为true ifTrue时返回元素flag,ifFalse在flagfalse 时返回元素.

或者,在matlab中,我可以用循环来做到这一点:

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

有矢量化的方法吗？

谢谢

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)