2011-10-21 55 views

回答

19

您正在尋找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,等來的值)的概括:numpy.choose()

+0

就是這樣。謝謝。選擇似乎很複雜,但我會在某處找到它的用處。 – hatmatrix