one line if else python