Create Numpy Array With List Comprehension With If Statement