Taking Input Array In Python Geeks