python for in if