python for in if statement