Ternary operator for NumPy ndarray?
You are looking for numpy.where():
>>> print numpy.where(numpy.arange(10) < 3, 'a', 'b')['a', 'a', 'a', 'b', 'b', 'b', 'b', 'b', 'b', 'b']NumPy even has a generalization (that maps 0, 1, 2, etc. to values, instead of mapping only True and False): numpy.choose().