在python中的散点图中基于单击的Artist更改参数

时间:2018-06-13 19:25:46

标签: python matplotlib

我在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值,但我无法访问它。

有什么建议吗?

1 个答案:

答案 0 :(得分:0)

你可以尝试扩展像

这样的东西
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()

这个想法是基于这样一个事实,即你可以将你挑选的artistscatter函数返回的对象进行比较。