I want to reproduce plots like this one in matplotlib. I'm performing simulations where the displacement of a particle whose energy is defined by a function, e.g. E(X)=1/2kx^2. I have a list with the locations of the particle and want to plot the transitions over a plot of the energy function, preferably using arrows between successive locations. How do I do this in matplotlib?

I have attached an image of a sample plot (a gradient descent plot - my requirement is quite similar).

So given a list E made from E = [(x*0.1-2)**2 1 for x in range(40)] and a list of points X = [0, 3, 1.5, 2.2, 1.9, 1.95, 2.05] I would like a plot like the one I've attached

CodePudding user response：

You can use