这是我想避免的解决方案。我在问题评论中链接的旧问题中看到有人提到这是一个np完全问题,但我想指出的是这并不重要。我测试了多达26个注释,这花费了几秒钟,但没有更多。任何实际情节都不会包含1000个注释。
注意事项:
如前所述,这不是超快的。 具体来说,我希望可以避免使用draw()。现在可以,只画两次。- 此代码允许仅在特定的正交(左/右/上/下)方向上添加任何新注释,但是可以对其进行扩展。
- 箭头的位置 取决于窗口 。这意味着 确保 注释后(带有箭头) 窗口大小或轴不会改变 。如果调整大小,请重新注释。
- 不是动态的,请参阅第3点。
背景:
Iplot
是一个帮助器类,我必须处理多图图形,处理着色,调整大小和现在进行注释。plt
是matplotlib.pyplot
- 此方法处理多个注释(或单个注释),现在可以解决冲突。
- 您可能已经猜到了,
Iplot.axes
握住我的斧头数字。
编辑
我删除了我的班级代码,以使其更可粘贴。应该给该函数指定轴,并且kwargs接受现有的box关键字以考虑先前的注释,该注释已在适当位置进行了编辑。注意我使用一个类来封装它。如果是第一次调用,该函数还必须返回要使用的框。
编辑2
过一会儿加快速度-无需绘制太多图像,最好循环两次,然后在两者之间更新渲染器。
编码:
def annotate(axes,boxes,labels,data,**kwargs): #slide should be relevant edge of bbox - e.g. (0,0) for left, (0,1) for bottom ... try: slide = kwargs.pop("slide") except KeyError: slide = None try: xytexts = kwargs.pop("xytexts") xytext = xytexts except KeyError: xytext = (0,2) xytexts = None try: boxes = kwargs.pop("boxes") except KeyError: boxes = list() pixel_diff = 1 newlabs = [] for i in range(len(labels)): try: len(xytexts[i]) xytext = xytexts[i] except TypeError: pass a = axes.annotate(labels[i],xy=data[i],textcoords='offset pixels', xytext=xytext,**kwargs) newlabs.append(a) plt.draw() for i in range(len(labels)): cbox = a.get_window_extent() if slide is not None: direct = int((slide[0] - 0.5)*2) current = -direct*float("inf") arrow = False while True: overlaps = False count = 0 for box in boxes: if cbox.overlaps(box): if direct*box.get_points()[slide] > direct*current: overlaps = True current = box.get_points()[slide] shift = direct*(current - cbox.get_points()[1-slide[0],slide[1]]) if not overlaps: break arrow = True position = array(a.get_position()) position[slide[1]] += shift * direct * pixel_diff a.set_position(position) plt.draw() cbox = a.get_window_extent() x,y = axes.transData.inverted().transform(cbox)[0] if arrow: axes.arrow(x,y,data[i][0]-x,data[i][1]-y,head_length=0,head_width=0) boxes.append(cbox) plt.draw() return boxes
任何改进的建议 都会受到热烈欢迎。非常感谢!
欢迎分享,转载请注明来源:内存溢出
评论列表(0条)