使用matplotlib在3D图形上交互式选择点

import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D

fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')

# 绘制点云
x = [1, 2, 3, 4, 5]
y = [2, 3, 4, 5, 6]
z = [3, 4, 5, 6, 7]
ax.scatter(x, y, z)

# 设置坐标轴标签
ax.set_xlabel('X')
ax.set_ylabel('Y')
ax.set_zlabel('Z')

# 显示图形
plt.show()

def on_pick(event):
    # 获取选中的点的索引
    ind = event.ind
    # 获取选中的点的坐标
    x_selected = x[ind]
    y_selected = y[ind]
    z_selected = z[ind]
    # 打印选中的点的坐标
    print(f"Selected point: ({x_selected}, {y_selected}, {z_selected})")

# 绑定事件处理函数
fig.canvas.mpl_connect('pick_event', on_pick)