python i for i in list if else