Numpy 中布尔索引的理解

array = [1,2,3,4,5,6]

con = array > 3 
# --> [False,False,False,True,True,True]

array[con] = 1
array[~con] = 0

# array -->[0,0,0,1,1,1]
 

上面的代码简单的举例了布尔索引的使用,他的核心是将 array > 3 以每个array中的元素是否满足运算条件而返回一个布尔数组。

发表评论

电子邮件地址不会被公开。 必填项已用*标注