matplotlib画动态图
1.画sin函数
import matplotlib.pyplot as plt
import numpy as np# Set up the figure and axis
fig, ax = plt.subplots()
xdata, ydata = [], []
line, = ax.plot([], [], lw=2) # 在坐标轴创建1个线对象，初始时没有数据，lw=2设置线宽为2
ax.set_xlim(0, 2 * np.pi)
ax.set_ylim(-1.5, 1.5)
ax.set_xlabel('x')
ax.set_ylabel('sin(x)')
ax.grid()"""
用于设置动画的初始状态。它将线的数据设置为空，并返回线对象。
"""
def init():line.set_data([], [])return line,
"""画曲线"""
def update(frame):xdata.append(frame / 10)ydata.append(np.sin(frame / 10))line.set_data(xdata, ydata)return line,# Animate
for i in range(100):update(i)plt.draw() # 绘制更新后的图形。plt.pause(0.1) # Pause for 100 ms to make the plot update visible
plt.show()
2.画散点图
import matplotlib.pyplot as plt
import numpy as np
import time# Set up the figure and axis
fig, ax = plt.subplots()
xdata, ydata = [], []
scat = ax.scatter([], [])
ax.set_xlim(-1, 1)
ax.set_ylim(-1, 1)
ax.set_xlabel('X')
ax.set_ylabel('Y')def update(frame):# 生成一个-1到1之间的随机浮点数xdata.append(np.random.uniform(-1, 1))ydata.append(np.random.uniform(-1, 1))scat.set_offsets(np.c_[xdata, ydata])return scat,# Animate
for i in range(100):update(i)plt.draw()plt.pause(0.1) # Pause for 100 ms to make the plot update visibleplt.show()
3.画折线图
import matplotlib.pyplot as plt
import numpy as np
import time# Set up the figure and axis
fig, ax = plt.subplots()
epochs = []
losses = []
line, = ax.plot([], [], lw=2)
ax.set_xlim(0, 100)
ax.set_ylim(0, 1)
ax.set_xlabel('Epoch')
ax.set_ylabel('Loss')
ax.grid()def update(epoch, loss):epochs.append(epoch)losses.append(loss)line.set_data(epochs, losses)ax.relim()ax.autoscale_view()return line,# Simulate updating loss over epochs
for epoch in range(100):fake_loss = np.random.rand()update(epoch, fake_loss)plt.draw()plt.pause(0.1) # Pause for 100 ms to make the plot update visibleplt.show()
