python if return 一行