if else using ternary operator in python