python if else 写 一行