1. 程式人生 > >將int型別的陣列轉換為bool陣列 in Python

將int型別的陣列轉換為bool陣列 in Python

#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)