Animate time series python

import numpy as np
import matplotlib.pyplot as plt
from matplotlib import animation

dt = 0.01
tfinal = 1
x0 = 0

sqrtdt = np.sqrt(dt)
n = int(tfinal/dt)
xtraj = np.zeros(n+1, float)
trange = np.linspace(start=0,stop=tfinal ,num=n+1)
xtraj[0] = x0

for i in range(n):
    xtraj[i+1] = xtraj[i] + np.random.normal()

x = trange
y = xtraj

# animation line plot example

fig, ax = plt.subplots(1, 1, figsize = (6, 6))

def animate(i):
    ax.cla() # clear the previous image
    ax.plot(x[:i], y[:i]) # plot the line
    ax.set_xlim([x0, tfinal]) # fix the x axis
    ax.set_ylim([1.1*np.min(y), 1.1*np.max(y)]) # fix the y axis

anim = animation.FuncAnimation(fig, animate, frames = len(x) + 1, interval = 1, blit = False)
plt.show()

Source: stackoverflow.com

Similar codes

2 votes114 views1 answer
Django register models
Python
8 votes119 views2 answers
How do I set Conda to activate the base environment by default?
Python
0 vote107 views1 answer
Python ignore first value in generator
Python
1 vote146 views1 answer
Module turtle has no forward member
Python
0 vote118 views1 answer
[Solved]AttributeError: ‘numpy.ndarray’ object has no attribute ‘index’
Python
5 votes113 views1 answer
Python pip yaml
Python
0 vote117 views1 answer
Main() invalid syntax
Python
1 vote133 views1 answer
Error popup in django not visible
Python Django
1 vote105 views2 answers
Dropna in specific column pandas
Python
5 votes111 views5 answers
Fibonacci series program in python
Python