Matplotlib散点图动画

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

# 生成随机数据
np.random.seed(42)
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()
scatter = ax.scatter(x, y, c=colors, s=sizes, alpha=0.5)

# 更新数据的函数
def update(frame):
    # 生成新的随机数据
    x = np.random.rand(100)
    y = np.random.rand(100)
    colors = np.random.rand(100)
    sizes = 1000 * np.random.rand(100)
    
    # 更新散点图的数据
    scatter.set_offsets(np.c_[x, y])
    scatter.set_color(colors)
    scatter.set_sizes(sizes)

# 创建动画
animation = FuncAnimation(fig, update, frames=range(100), interval=200)

# 展示动画
plt.show()