Тернарный оператор для NumPy ndarray?

Question

Есть ли у NumPy троичный оператор?Например, в R есть векторизованная if-else функция:

> ifelse(1:10 < 3,"a","b")
 [1] "a" "a" "b" "b" "b" "b" "b" "b" "b" "b"

Есть ли что-нибудь эквивалентное в NumPy?

Answer 1

Вы ищете numpy.where():

>>> print numpy.where(numpy.arange(10) < 3, 'a', 'b')
['a', 'a', 'a', 'b', 'b', 'b', 'b', 'b', 'b', 'b']

NumPy даже имеет обобщение (которое отображает значения 0, 1, 2 и т. Д. Вместо отображения только True и False): numpy.choose().