【python/matplotlib】动态更新数据

说在前面

• python版本：3.7
• matploylib版本：3.4.2
• 环境：win10

场景

• 实时更新数据，可重绘
• 当前只测试了scatter，其它类型的暂未尝试

代码

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

np.random.seed(19680801)

fig, ax = plt.subplots()
x, y, s, c = [],[],[],[]
sc = ax.scatter(x,y,s,c=c)
plt.xlim(0,10)
plt.ylim(0,10)

def animate_append(i):
    x.append(np.random.rand(1)*10)
    y.append(np.random.rand(1)*10)
    s.append(np.random.rand(1)*100)
    c.append(np.random.rand(1))
    # set x and y
    sc.set_offsets(np.c_[x,y])
    # set size
    sc.set_sizes(np.array(s).ravel())
    # set color
    sc.set_array(np.array(c).ravel())

def animate(i):

    x.clear()
    y.clear()
    s.clear()
    c.clear()
    x.append(np.random.rand(1) * 10)
    y.append(np.random.rand(1) * 10)
    s.append(np.random.rand(1) * 100)
    c.append(np.random.rand(3,))
    # set x and y
    sc.set_offsets(np.c_[x, y])
    # set size
    sc.set_sizes(np.array(s).ravel())
    # set color
    sc.set_array(np.array(c).ravel())

ani = matplotlib.animation.FuncAnimation(fig, animate,
                frames=2, interval=1000, repeat=True)
plt.show()

结果(gif)