Python Create List Using List Comprehension With If Statement