Animated 2D plot in Matplotlib

test3.py

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

# Create the figure and the axes
fig = plt.figure()
ax = fig.add_subplot(111, title = 'Sine wave animation')

# Initialize the axes
ax.set_xlim(0, 2*np.pi)
ax.set_ylim(-1, 1)

# Prepare the datapoints
x = np.linspace(0, 2*np.pi, 100)
y = np.sin(x)

bundled_data = [(x[i], y[i]) for i in range(0, 100)]

x_data, y_data = [], []

# Plot the data points
lines = ax.plot(x_data, y_data, animated = True)

# Setup the animation
def update(frame):
    x_point, y_point = frame
    x_data.append(x_point)
    y_data.append(y_point)
    lines[0].set_data(x_data, y_data)
    return lines

animation = FuncAnimation(fig, update, frames = bundled_data, blit = True, interval = 40)

# Display the plot
plt.show()