From 8db1f12d8bba84c697b1bf45204631343d41b66b Mon Sep 17 00:00:00 2001 From: "W. Trevor King" Date: Sat, 21 Jan 2012 13:53:14 -0500 Subject: [PATCH] Remove pylab.ion() from plot_image.py's main(). When pylab is in interactive mode, showing the figures does not halt script exectution. This means that the script will likely complete before the user has time to appreciate the figure (unless the are *really* fast ;)). Also use figure/axes methods where possible (instead of pylab.axes, etc.). It's cleaner that way ;). --- src/plot_image/plot_image.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/src/plot_image/plot_image.py b/src/plot_image/plot_image.py index 9eaaa20..591a7f4 100755 --- a/src/plot_image/plot_image.py +++ b/src/plot_image/plot_image.py @@ -149,9 +149,9 @@ def plot(X, Y, Z, full=False, title=None, contours=None, cmap=None): fig = pylab.figure() if full: - axes = pylab.axes([0, 0, 1, 1]) + axes = fig.add_axes([0, 0, 1, 1]) else: - axes = pylab.axes() + axes = fig.add_subplot(1, 1, 1) if title: axes.set_title(title) axes.set_axis_off() @@ -159,7 +159,7 @@ def plot(X, Y, Z, full=False, title=None, contours=None, cmap=None): if contours: cset = axes.contour(X[:-1,:-1], Y[:-1,:-1], Z, contours, cmap=cmap) # [:-1,:-1] to strip dummy last row & column from X&Y. - pylab.clabel(cset, inline=1, fmt='%1.1f', fontsize=10) + axes.clabel(cset, inline=1, fmt='%1.1f', fontsize=10) else: # pcolor() is much slower than imshow. #plot = axes.pcolor(X, Y, Z, cmap=cmap, edgecolors='none') @@ -279,7 +279,6 @@ def main(argv=None): if options.output: fig.savefig(options.output) else: - pylab.ion() pylab.show() -- 2.26.2