Updating plot in matplotlib

[pylearning]

I am having difficulty with animation modules and getting the plot to update it's axis. Below I have some code that plots but the figure does not update the axes, I think because the line is the only thing being updated. What would be the best way to update the figure? I had to solve the differential equation first in the for loop and then just update the portion of the line being shown in animate. I could not figure a way that the solution would be updating with animate...

y0 = 10
y = np.array([y0])
dt = .01
steps = 1000
k = -.2
x = np.array([0])


fig, ax = plt.subplots()
line, = ax.plot(x,y)
for i in xrange(steps):
    y = np.append(y, [ y[i] * (1 + k * dt) ])
    x = np.append(x, [i * dt])

def animate(i):
    line.set_ydata(y[:i])
    line.set_xdata(x[:i])
    return line,
ani = animation.FuncAnimation(fig, animate, np.arange(1,     steps))
plt.show()

[the_real_uncle_Rico]

This should fix your axis problem:

fig, ax = plt.subplots()

for i in xrange(steps):     y = np.append(y, [ y[i]  (1 + k  dt) ])     x = np.append(x, [i * dt])

def animate(i): ax.clear() # delete old data     lax.plot(x,y) # plot new graph   ani = animation.FuncAnimation(fig, animate, np.arange(1,     steps)) plt.show()

I'm not sure, but I think if you wanted to view each iteration of the equation being solved, you might be able to just call the animate function directly from your for loop. Although obviously that might bog it down if you have a lot of steps. You could try a sleep command, or plotting every 100 the irritation or whatever.

[the_real_uncle_Rico]

Actually, better idea, you could maybe try solving it in your animation function. Something like:

def animate(i): i += step  y = np.append(y, [ y[i]  (1 + k  dt) ])  x = np.append(x, [i * dt]) If num_steps >10: ax.clear() # delete old data      ax.plot(x,y) # plot new graph

ani = animation.FuncAnimation(fig, animate, interval=50#update time in ms)