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)