python for if else 一行