python3 if else one liner