can we write if/else into one line in python?