python if 一行寫法