Numpy And Matplotlib Tutorial Geeksforgeeks Problem