can we use if else in list comprehension python