python for in if in one line