Toán tử Ternary cho NumPy ndarray?

Question

NumPy có toán tử bậc ba không? Ví dụ: trong R, có chức năng if-else được vector hóa:

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

Có điều gì tương đương trong NumPy không?

Answer 1

Bạn đang tìm kiếm numpy.where():

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

NumPy thậm chí còn có một sự tổng quát (mà bản đồ 0, 1, 2, vv để các giá trị, thay vì lập bản đồ chỉ True và False): numpy.choose().