python list comprehension double if