Take User Input Array In Python Geeksforgeeks