Python Numpy Gradient Descent Visualization