python numpy.array中的True,False 转为1,0

numpy中对于一个数组

[True,False]

 

1.转为int，最简单的方式就是后面加0

import numpy as np
a = np.array([True, False])
#print(a)
b = a + 0
#print(b)

注意：如果是对图片数组这样操作，会改变图片中数据的类型，调用某些opencv中的函数时会报错，

这时可以尝试将数组类型转换一下，如下：

import numpy as np
img = img + 0
img = np.uint8(img)

 

2.那么如何把0,1转成True,false呢？用>或<，即可得到bool类型

import numpy as np
a = np.array([1, 0])
#print(a)
b = a > 0
print(b)

 

如果是numpy.array类型的，则支持上面的操作。

如果是单纯的数组类型，是不行的，可以使用map的方式。

也可以把数组转为numpy的数组再用上面的方法处理。

def fun(x):
    if x:
        return True
    return False

a = [1, 0]
#print(a)
b = list(map(fun,a))
print(b)