python-如何在numpy上映射函数，并在每个变量上设置条件？

我在尝试在numpy数组上映射此函数时遇到此错误：

>>> a = np.array([1, 2, 3, 4, 5])
>>> g = lambda x: 0 if x % 2 == 0 else 1
>>> g(a)
Traceback (most recent call last):
  File "", line 1, in 
  File "", line 1, in 
ValueError: The truth value of an array with more than one element is ambiguous. Use a.any() or a.all()     

我期待结果array([ 1,  0,  1, 0, 1])

在这种情况下工作正常时：

>>> f = lambda x: x ** 2
>>> f(a)
array([ 1,  4,  9, 16, 25])

我该怎么做才能将函数g映射到数组上a比for循环更快，最好使用numpy更快的代码？