numpy将bool值转换成数值

转载请注明出处
个人博客:https://maxusun.github.io/

在实际开发中我们需要将bool值转换成0/1或者其他数值，这里整理了两种方法。

使用np.array().astype(int)

a = np.array([False,True,False])
int_a = a.astype(int)

使用np.where()

a = np.array([False,True,False])
a = np.where(a,1,-1)        # 得到 a [-1,1,1]

解释一下np.where函数：

常见的两种用法：

• 第一种

a = np.array([1,2,3,4,5,6])
b = np.where(a > 3)                         # 返回一个和a大小相同的array，满足条件的位置赋值True，不满足的赋值False
# b = [False,False,False,True,True,True]

• 第二种

a = np.array([1,2,3,4,5,6])
b = np.where(a>3,1,-1)                      # 三个参数，第一个是同上面一样的判断条件，第二个：满足判断条件位置的赋值。第三个：不满足条件位置的赋值
# b = [-1,-1,-1,1,1,1]