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)

 

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的方式

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

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