python for in if 一行