General vectorization of if-else tests

def f(x):                     # scalar x
    if condition:
        x = <expression1>
    else:
        x = <expression2>
    return x

def f_vectorized(x):          # scalar or array x
    x1 = <expression1>
    x2 = <expression2>
    return where(condition, x1, x2)