高效的Matplotlib重绘

Question

我使用Matplotlib允许用户使用鼠标点击选择有趣的数据点，使用与this answer.非常相似的方法

实际上，散热图显示在热图图像上，鼠标点击可以添加或删除散点图。

我的数据是使用pcolormesh()在后​​台绘制的，因此当我使用axis.figure.canvas.draw()更新画布时，会重绘散点图和背景热图。鉴于热图的大小，这对于可用的接口来说太慢了。

有没有一种方法可以选择性地重新绘制散点而不用重绘背景？

示例代码：

points = []  # Holds a list of (x,y) scatter points 

def onclick(event):
    # Click event handler to add points
    points.append( (event.x, event.y) )
    ax.figure.canvas.draw()

fig = plt.figure()
ax = plt.figure()
# Plot the background
ax.pcolormesh(heatmap_data)

# Add the listener to handle clicks
cid = fig.canvas.mpl_connect("button_press_event", onclick)
plt.show()

Answer 1

当然！你想要的是blitting。如果您没有编写gui，可以使用matplotlib.animation简化其中的一部分，但如果您希望事物是交互式的，则需要直接处理它。

在matplotlib术语中，您需要fig.canvas.copy_from_bbox的组合，然后轮流调用fig.canvas.restore_region(background)，ax.draw_artist(what_you_want_to_draw)和fig.canvas.blit：

background = fig.canvas.copy_from_bbox(ax.bbox)

for x, y in user_interactions:
    fig.canvas.restore_region(background)
    points.append([x, y])
    scatter.set_offsets(points)
    ax.draw_artist(scatter)
    fig.canvas.blit(ax.bbox)

---

简单Blitting示例：添加点

在您的情况下，如果您只是添加积分，则实际上可以跳过保存并恢复背景。但是，如果你走这条路线，由于抗锯齿点重复地重新绘制在一起，你将最终对绘图进行一些细微的改变。

无论如何，这是你想要的事物类型的最简单的例子。这只涉及添加点，并跳过保存和恢复背景，如上所述：

import matplotlib.pyplot as plt
import numpy as np

def main():
    fig, ax = plt.subplots()
    ax.pcolormesh(np.random.random((100, 100)), cmap='gray')

    ClickToDrawPoints(ax).show()

class ClickToDrawPoints(object):
    def __init__(self, ax):
        self.ax = ax
        self.fig = ax.figure
        self.xy = []
        self.points = ax.scatter([], [], s=200, color='red', picker=20)
        self.fig.canvas.mpl_connect('button_press_event', self.on_click)

    def on_click(self, event):
        if event.inaxes is None:
            return
        self.xy.append([event.xdata, event.ydata])
        self.points.set_offsets(self.xy)
        self.ax.draw_artist(self.points)
        self.fig.canvas.blit(self.ax.bbox)

    def show(self):
        plt.show()

main()

---

有时简单太简单了

但是，假设我们想要右键单击删除一个点。

在这种情况下，我们需要能够在不重绘背景的情况下恢复背景。

好的，一切都很好。我们将使用类似于我在答案顶部提到的伪代码片段的内容。

然而，有一点需要注意：如果调整了数字，我们需要更新背景。同样，如果轴以交互方式缩放/平移，我们需要更新背景。基本上，您需要在绘制绘图时随时更新背景。

很快你需要变得相当复杂。

---

更复杂：添加/拖动/删除点

以下是您最终实施的“脚手架”的一般示例。

这有点效率低，因为情节会被绘制两次。 （例如，平移将慢）。有可能绕过那个，但我会把这些例子留下来。

这实现了添加点，拖动点和删除点。要在交互式缩放/平移后添加/拖动点，请再次单击工具栏上的缩放/平移工具以禁用它们。

这是一个相当复杂的例子，但希望它能让人们了解一下通常用于交互式绘制/拖动/编辑/删除matplotlib艺术家的框架类型，而无需重新绘制整个图。

import numpy as np
import matplotlib.pyplot as plt

class DrawDragPoints(object):
    """
    Demonstrates a basic example of the "scaffolding" you need to efficiently
    blit drawable/draggable/deleteable artists on top of a background.
    """
    def __init__(self):
        self.fig, self.ax = self.setup_axes()
        self.xy = []
        self.tolerance = 10
        self._num_clicks = 0

        # The artist we'll be modifying...
        self.points = self.ax.scatter([], [], s=200, color='red',
                                      picker=self.tolerance, animated=True)

        connect = self.fig.canvas.mpl_connect
        connect('button_press_event', self.on_click)
        self.draw_cid = connect('draw_event', self.grab_background)

    def setup_axes(self):
        """Setup the figure/axes and plot any background artists."""
        fig, ax = plt.subplots()

        # imshow would be _much_ faster in this case, but let's deliberately
        # use something slow...
        ax.pcolormesh(np.random.random((1000, 1000)), cmap='gray')

        ax.set_title('Left click to add/drag a point\nRight-click to delete')
        return fig, ax

    def on_click(self, event):
        """Decide whether to add, delete, or drag a point."""
        # If we're using a tool on the toolbar, don't add/draw a point...
        if self.fig.canvas.toolbar._active is not None:
            return

        contains, info = self.points.contains(event)
        if contains:
            i = info['ind'][0]
            if event.button == 1:
                self.start_drag(i)
            elif event.button == 3:
                self.delete_point(i)
        else:
            self.add_point(event)

    def update(self):
        """Update the artist for any changes to self.xy."""
        self.points.set_offsets(self.xy)
        self.blit()

    def add_point(self, event):
        self.xy.append([event.xdata, event.ydata])
        self.update()

    def delete_point(self, i):
        self.xy.pop(i)
        self.update()

    def start_drag(self, i):
        """Bind mouse motion to updating a particular point."""
        self.drag_i = i
        connect = self.fig.canvas.mpl_connect
        cid1 = connect('motion_notify_event', self.drag_update)
        cid2 = connect('button_release_event', self.end_drag)
        self.drag_cids = [cid1, cid2]

    def drag_update(self, event):
        """Update a point that's being moved interactively."""
        self.xy[self.drag_i] = [event.xdata, event.ydata]
        self.update()

    def end_drag(self, event):
        """End the binding of mouse motion to a particular point."""
        for cid in self.drag_cids:
            self.fig.canvas.mpl_disconnect(cid)

    def safe_draw(self):
        """Temporarily disconnect the draw_event callback to avoid recursion"""
        canvas = self.fig.canvas
        canvas.mpl_disconnect(self.draw_cid)
        canvas.draw()
        self.draw_cid = canvas.mpl_connect('draw_event', self.grab_background)

    def grab_background(self, event=None):
        """
        When the figure is resized, hide the points, draw everything,
        and update the background.
        """
        self.points.set_visible(False)
        self.safe_draw()

        # With most backends (e.g. TkAgg), we could grab (and refresh, in
        # self.blit) self.ax.bbox instead of self.fig.bbox, but Qt4Agg, and
        # some others, requires us to update the _full_ canvas, instead.
        self.background = self.fig.canvas.copy_from_bbox(self.fig.bbox)

        self.points.set_visible(True)
        self.blit()

    def blit(self):
        """
        Efficiently update the figure, without needing to redraw the
        "background" artists.
        """
        self.fig.canvas.restore_region(self.background)
        self.ax.draw_artist(self.points)
        self.fig.canvas.blit(self.fig.bbox)

    def show(self):
        plt.show()

DrawDragPoints().show()