python - 加快在 matplotlib 中绘制图像

Question

我在 Spyder IDE 中编写了一些 Python 来并排绘制一对图像，以便我可以直观地检查它们。大多数时候我只需要 3 秒钟就能看到它们，但偶尔我需要更长的时间才能仔细观察。因此，我没有使用 time.sleep，而是将其编码为等待我按下 Enter 键，如下所示:

import matplotlib.pyplot as plt
import os

def VI_segmentation():
    root = os.getcwd()
    NR_dir = root + '\\Neurite_Results\\'
    SO_dir = root + '\\Segmentation_Overlays\\'
    jpgs = os.listdir(NR_dir)
    fig = plt.figure(figsize=(20,12))
    for jpg in jpgs:
        fig.suptitle(jpg , fontsize=14, fontweight='bold')
        image_NR = plt.imread(NR_dir + jpg)
        image_SO = plt.imread(SO_dir + jpg)
        plt.subplot(121)
        plt.imshow(image_NR)
        plt.subplot(122)
        plt.imshow(image_SO)
        plt.draw()
        plt.pause(0.01)

        input('Press Enter to continue')

VI_segmentation()

问题是我的思考速度比我的电脑还快 :)。计算机需要 5 或 6 秒才能响应 Enter 键，并在响应后再花几秒钟进行更新。在浏览数百张大部分都很好的图像时，会导致糟糕的人体工程学。任何简化此代码的想法都将不胜感激。

Answer 1

这个版本的代码最终解决了我的问题:

import matplotlib.pyplot as plt
import os

def VI_segmentation():
    plt.ion()
    root = os.getcwd()
    NR_dir = root + '\\Neurite_Results\\'
    SO_dir = root + '\\Segmentation_Overlays\\'
    jpgs = os.listdir(NR_dir)
    f = plt.figure(figsize=(22,12))
    ax1 = f.add_subplot(121)
    ax2 = f.add_subplot(122)
    image_NR = plt.imread(NR_dir + jpgs[0])
    image_SO = plt.imread(SO_dir + jpgs[0])
    im1 = ax1.imshow(image_NR)
    im2 = ax2.imshow(image_SO) 
    f.suptitle(jpgs[0] , fontsize=14, fontweight='bold')
    f.show()
    plt.pause(0.01)
    input('Press Enter to continue')

    for jpg in jpgs[1:]:
        f.suptitle(jpg , fontsize=14, fontweight='bold')
        image_NR = plt.imread(NR_dir + jpg)
        image_SO = plt.imread(SO_dir + jpg)
        im1.set_data(image_NR)
        im2.set_data(image_SO)
        f.canvas.draw()
        plt.pause(0.01)
        input('Press Enter to continue')

VI_segmentation()

关键是更改绘图中的数据，而不是添加新绘图。这个答案对我很有帮助。

Why does my pylab animation slow down with each update?

奇怪的是，当我开始更改绘图数据而不是重新绘图时，我开始出现奇怪的行为，即图形会放大但周围的窗口不会。不知何故，这个 fig.set_size_inches 被破坏了，所以我移动了图形创建和轴创建，这样我就可以在制作图形时设置图形大小。