将int类型的数组转换为bool数组 in Python

版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/xuluhui123/article/details/80948991

#coding=UTF-8

import numpy as np

# int array
array1 = np.array([1,0,1,0])
# convert int array to bool list
list1 = [True if array1[i]==0 else False for i in range(len(array1))]
# convert bool list to bool array
array1 = np.array(list1)

print(array1)

#array1 = np.array([True,False,True,False])

array2 = np.array([[1,2,3,4],[5,6,7,8],[9,10,11,12],[13,14,15,16]])

# mask 
res = array2[array1,:]

print(res)