使用滑块更新matplotlib中的随机粒子

import numpy as np
import matplotlib.pyplot as plt
from matplotlib.widgets import Slider

fig, ax = plt.subplots()
plt.subplots_adjust(left=0.1, bottom=0.25)  # 调整图形位置

# 生成随机粒子的初始位置
num_particles = 100
x = np.random.rand(num_particles)
y = np.random.rand(num_particles)

# 绘制随机粒子
scatter = ax.scatter(x, y)
ax.set_xlim(0, 1)  # 设置x轴范围
ax.set_ylim(0, 1)  # 设置y轴范围
ax.set_aspect('equal')  # 设置x和y轴的比例相等
ax.set_title('Random Particles')  # 设置图形标题

# 创建滑块的位置和大小
slider_ax = plt.axes([0.1, 0.1, 0.8, 0.05])
slider = Slider(slider_ax, 'Slider', 0, 1, valinit=0.5)

# 定义滑块更新函数
def update(val):
    # 获取滑块的值
    value = slider.val
    
    # 更新粒子的位置
    x_new = x + np.random.normal(0, value, num_particles)
    y_new = y + np.random.normal(0, value, num_particles)
    
    # 更新粒子的位置数据
    scatter.set_offsets(np.column_stack((x_new, y_new)))
    
    # 重新绘制图形
    fig.canvas.draw_idle()

# 绑定滑块的更新函数
slider.on_changed(update)

# 显示图形
plt.show()