使用两帧动画制作matplotlib散点图

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

# 创建随机数据
np.random.seed(0)
x = np.random.rand(100)
y = np.random.rand(100)
colors = np.random.rand(100)
sizes = 1000 * np.random.rand(100)

fig, ax = plt.subplots()

sc = ax.scatter(x, y, c=colors, s=sizes, alpha=0.5)

def update(frame):
    # 更新散点图的位置和颜色
    sc.set_offsets(np.random.rand(100, 2))
    sc.set_color(np.random.rand(100))

ani = animation.FuncAnimation(fig, update, frames=2, interval=1000)

plt.show()