加快在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()

问题是我的思维速度比我的计算机快:).计算机响应Enter键需要5或6秒钟,响应后需要几秒钟才能更新.通过数百张大多数精细的图像来制作糟糕的人体工程学设计.任何简化此代码的想法都将非常感激.

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()

关键是改变图中的数据,而不是添加新的图.这个答案对我很有帮助.

为什么每次更新时我的pylab动画都会变慢？

奇怪的是,当我开始更改绘图数据而不是重新绘制时,我开始变得奇怪的行为,其中数字会放大,但周围的窗口不会.不知何故,这个fig.set_size_inches被打破了所以我移动了图形创建和轴创建,所以我可以设置图形时的图形大小.