#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)
将int类型的数组转换为bool数组 in Python
最新推荐文章于 2024-07-20 22:14:27 发布