How do I illustrate a 3D graph of gradient descent linear regression using matplotlib?

Question

I am having trouble with plotting a 3d graph for gradient descent using python\'s matplotlib. The commented code in the gradient_descent function was what I tried but doesn\