python for in if loop