如何通过按钮控件更新Matplotlib散点图的底层数据

你的代码核心问题是**set_offsets()的参数格式错误**，以及函数里的数据集对应错误，修正后就能正常切换坐标了：

错误原因 #

• PathCollection.set_offsets() 要求传入的是一个N行2列的二维数组，每一行对应一个点的(x,y)坐标，而不是分开传入x和y两个一维数组。
• 原函数set_data1里错误地写了x1, y2，应该对应数据集1的x1, y1。

修正后的完整代码 #

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

x1,y1 = np.random.normal(0,1, (2,20)) # dataset 1
x2,y2 = np.random.normal(0,1, (2,20)) # dataset 2

# 创建画布和坐标轴
fig, ax = plt.subplots()
p = ax.scatter(x1, y1)

# 定义切换数据集的函数
def set_data1(event):
    # 将x1和y1组合成N×2的数组
    offsets = np.column_stack((x1, y1))
    p.set_offsets(offsets)
    # 调整坐标轴范围以适配新数据
    ax.relim()
    ax.autoscale_view()
    plt.draw()
    
def set_data2(event):
    offsets = np.column_stack((x2, y2))
    p.set_offsets(offsets)
    ax.relim()
    ax.autoscale_view()
    plt.draw()

# 创建按钮位置和按钮
btn_1_loc = fig.add_axes([0.02, 0.90, 0.2, 0.075])
btn_2_loc = fig.add_axes([0.02, 0.80, 0.2, 0.075])

btn_1 = Button(btn_1_loc, '显示数据集1')
btn_1.on_clicked(set_data1)

btn_2 = Button(btn_2_loc, '显示数据集2')
btn_2.on_clicked(set_data2)

plt.show()

关键修改点 #

• 使用np.column_stack()将x和y数组合并成符合要求的二维数组，传给set_offsets()。
• 修复了set_data1里的数据集对应错误，改为正确的x1, y1。
• 添加了ax.relim()和ax.autoscale_view()，确保切换数据后坐标轴能自动适配新的点范围（提升交互体验）。

内容的提问来源于stack exchange，提问作者Mitchell van Zuylen