Single Line If Statement In Python