One Line Python If Statement