Python If In One Line