Numpy的array的布尔型索引与赋值

names=np.array(['Bob','joe','will','haha'])

data=np.randn(2,2)

1、可使用names==‘Bob’来判断array中的值是否与Bob相同：

>>>names=='Bob'
array([ True, False, False, False], dtype=bool)

2、可使用names!='Bob'来判断array中的值是否与Bob相同

如需判断多个条件可使用 | 或者 & ：

mask=(names=='Bob')|(names=='will')

3、通过data<0等条件可以将data里所有负值设置为0

data[data<0]=0
data
matrix([[ 0.        ,  0.        ,  0.63229813,  0.04533078]])