np.where 允许您选择要为布尔类型查询分配的值,例如
test = [0,1,2]
np.where(test==0,'True','False')
print test
['True','False','False']
Run Code Online (Sandbox Code Playgroud)
这基本上是一个“if”语句。对于 numpy 数组,是否有一种Python式的方法来使用“if,else if,else”类型的语句(具有不同的情况)?
这是我的解决方法:
color = [0,1,2]
color = np.where(color==0,'red',color)
color = np.where(color==1,'blue',color)
color = np.where(color==2,'green',color)
print color
['red','blue','green']
Run Code Online (Sandbox Code Playgroud)
但我想知道是否有更好的方法来做到这一点。