python list comprehension two ifs