我尝试使用DataCursor方法(https://stackoverflow.com/a/4674445/1301710)使用matplotlib标记点。我有几千个点,我想看看他们的标签上的鼠标。但是,有两个不同点:一个是我绘制散点图,二是我要为每个点标记名称,而不仅仅是x,y坐标。在
这是我的密码import os
import matplotlib.pyplot as plt
class DataCursor(object):
text_template = 'x: %0.2f\ny: %0.2f'
x, y = 0.0, 0.0
xoffset, yoffset = -20, 20
text_template = 'x: %0.2f\ny: %0.2f'
def __init__(self, ax, labels,x,y):
self.ax = ax
self.xlist = x
self.ylist = y
self.labels = labels
self.annotation = ax.annotate(self.text_template,
xy=(self.x, self.y), xytext=(self.xoffset, self.yoffset),
textcoords='offset points', ha='right', va='bottom',
bbox=dict(boxstyle='round,pad=0.5', fc='yellow', alpha=0.5),
arrowprops=dict(arrowstyle='->', connectionstyle='arc3,rad=0')
)
self.annotation.set_visible(False)
def __call__(self, event):
self.event = event
xdata, ydata = event.artist.get_data()
#self.x, self.y = xdata[event.ind], ydata[event.ind]
self.x, self.y = event.mouseevent.xdata, event.mouseevent.ydata
self.label = self.labels[self.xlist.index(self.x)]
if self.x is not None:
self.annotation.xy = self.x, self.y
self.annotation.set_text(self.label)
self.annotation.set_visible(True)
event.canvas.draw()
def process():
#code to make ht_dict here
# ht_dict has the following format: 'ht1' = [nov14count, nov21count] where each key is a string and each value is a list of two integers
print("Start making scatter plot..")
hts = []
nov14 = []
nov21 = []
for key in ht_dict.keys():
nov14.append(ht_dict[key][0])
nov21.append(ht_dict[key][1])
hts.append(key)
fig = plt.figure()
scatter = plt.scatter(nov14, nov21)
fig.canvas.mpl_connect('pick_event', DataCursor(plt.gca(), hts, nov14, nov21))
scatter.set_picker(5)
plt.show()
process()
但我得到了以下错误:
^{pr2}$
我希望能够在mouseover上看到存储在list hts中的字符串,分别位于nov14和nov21列表中相同索引下的x和y坐标。我不知道该怎么解释这个错误,希望你能帮我。
我的另一个问题是(在DataCursor线程中尝试对现有绘图进行更改)使用index of取回label会给我一个list error中不存在的值,因为单击的值可能与列表中的值不完全相同。你有什么更好的方法来显示某个点的标签/名称吗?在
如果能给我提供任何指导或参考,我将不胜感激。在
谢谢!在