Taking An Integer Input In Python Definition