python conditional statement one line