首页 > 解决方案 > 在python的散点图中根据单击的艺术家更改参数

问题描述

我在python中有一个带有选项的散点图picker=True

现在我想根据我选择的点打开另一个带有参数的 python 脚本。

我确实有一个工作活动

self.fig.canvas.mpl_connect('pick_event', self.onpick)
def onpick(self, event):
    artist = event.artist
    print("Hello")
...
# The scatter plot is created with
plt.scatter(data[:, 0], data[:, 1], cmap=cmap, c=data[:, 2], s=100, picker=True)

总是当我点击一个散点 python 打印Hello

现在我想为每个艺术家添加一个简单的信息(例如他们被创建为的顺序int),所以当我点击它时,我可以访问这些信息并将它用作我想要在点击时打开的 python 脚本的参数.

另一个(更糟糕的)想法是,获取该艺术家在我的情节中的位置,然后根据该艺术家的位置计算我的参数。

可悲的是,如果其中一个想法是可能的,我找不到任何信息。我在https://matplotlib.org/2.0.0/users/artists.html发现艺术家有一个 x 和一个 y 值,但我无法访问它。

有什么建议么?

标签: pythonmatplotlib

解决方案


您可以尝试扩展类似

import matplotlib.pyplot as plt

def onpick(event):
    artist = event.artist
    for scatter_plot in scatter_plots:
        if artist is scatter_plot.plot:
            print(scatter_plot.order)

class MyScatterPlot():
    def __init__(self, order):
        self.order = order
        self.plot = None

fig, ax = plt.subplots()

scatter_plot1 = MyScatterPlot("first")
scatter_plot2 = MyScatterPlot("second")

scatter_plot1.plot = ax.scatter([1,1,1],[1,2,3], picker=True)
scatter_plot2.plot = ax.scatter([3,3,3],[1,2,3], picker=True)

scatter_plots = [scatter_plot1, scatter_plot2]

fig.canvas.mpl_connect('pick_event', onpick)

fig.show()

这个想法是基于这样一个事实,即您可以将您选择的对象与函数artist返回的对象进行比较。scatter


推荐阅读