python for in if else 一行