Ternary operator dla NumPy ndarray?

Question

Czy NumPy ma operatora trójskładnikowego? Na przykład, w R jest wektorowy if-else funkcja:

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

Czy jest coś równoważne w NumPy?

Answer 1

Szukasz numpy.where():

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

NumPy ma nawet uogólnienie mapy (to 0, 1, 2, itd do wartości, zamiast mapowania tylko prawda i fałsz): numpy.choose().