python dictionary comprehension if else