python3 list comprehension if