我在matplotlib中的轴对象上绘制了一个图例,但声称将其置于智能位置的默认定位似乎不起作用。理想情况下,我希望用户可以拖动图例。怎么办呢?
答案 0 :(得分:29)
注意:现在内置于matplotlib
leg = plt.legend()
if leg:
leg.draggable()
将按预期工作
好吧,我发现解决方案的各个部分分散在邮件列表中。我想出了一个很好的模块化代码块,你可以放入并使用......这里是:
class DraggableLegend:
def __init__(self, legend):
self.legend = legend
self.gotLegend = False
legend.figure.canvas.mpl_connect('motion_notify_event', self.on_motion)
legend.figure.canvas.mpl_connect('pick_event', self.on_pick)
legend.figure.canvas.mpl_connect('button_release_event', self.on_release)
legend.set_picker(self.my_legend_picker)
def on_motion(self, evt):
if self.gotLegend:
dx = evt.x - self.mouse_x
dy = evt.y - self.mouse_y
loc_in_canvas = self.legend_x + dx, self.legend_y + dy
loc_in_norm_axes = self.legend.parent.transAxes.inverted().transform_point(loc_in_canvas)
self.legend._loc = tuple(loc_in_norm_axes)
self.legend.figure.canvas.draw()
def my_legend_picker(self, legend, evt):
return self.legend.legendPatch.contains(evt)
def on_pick(self, evt):
if evt.artist == self.legend:
bbox = self.legend.get_window_extent()
self.mouse_x = evt.mouseevent.x
self.mouse_y = evt.mouseevent.y
self.legend_x = bbox.xmin
self.legend_y = bbox.ymin
self.gotLegend = 1
def on_release(self, event):
if self.gotLegend:
self.gotLegend = False
......并在您的代码中......
def draw(self):
ax = self.figure.add_subplot(111)
scatter = ax.scatter(np.random.randn(100), np.random.randn(100))
legend = DraggableLegend(ax.legend())
我通过电子邮件发送了Matplotlib-users组,John Hunter非常友好地将我的解决方案添加到SVN HEAD。
2010年1月28日星期四下午3:02,亚当 弗雷泽 写道:
我认为自从那以后我就可以解决这个可拖延的传奇问题 我花了很长时间才能吸收所有分散的知识 邮件列表......
很酷 - 很好的例子。我添加了代码 legend.py。现在你可以做到
leg = ax.legend()
leg.draggable()启用可拖动模式。您可以 反复调用此函数进行切换 可拖延的国家。
我希望这对使用matplotlib的人有帮助。
答案 1 :(得分:14)
在较新版本的Matplotlib(v1.0.1)中,这是内置的。
def draw(self):
ax = self.figure.add_subplot(111)
scatter = ax.scatter(np.random.randn(100), np.random.randn(100))
legend = ax.legend()
legend.draggable(state=True)
如果您以交互方式使用matplotlib(例如,在IPython的pylab模式下)。
plot(range(10), range(10), label="test label")
plot(range(10), [5 for x in range(10)], label="another test")
l = legend()
l.draggable(True)
答案 2 :(得分:3)
甚至在较新的版本(3.0.2)中,它也已被弃用,并可能在将来的版本中表示属性(因此,它将无法调用)。
plot(range(10), range(10), label="test label")
plot(range(10), [5 for x in range(10)], label="another test")
plt.legend().set_draggable(True)