python if 一行