From 071f99600251dcf54b16c85f59054912ef0805a8 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Jostein=20B=C3=B8=20Fl=C3=B8ystad?= Date: Sat, 7 Apr 2012 16:57:56 +0200 Subject: [PATCH 2/2] Fix imsave to accept a third color/alpha dimension. --- lib/matplotlib/image.py | 4 ++-- 1 files changed, 2 insertions(+), 2 deletions(-) diff --git a/lib/matplotlib/image.py b/lib/matplotlib/image.py index d7b91da..5a7dce1 100644 --- a/lib/matplotlib/image.py +++ b/lib/matplotlib/image.py @@ -1217,7 +1217,7 @@ def imsave(fname, arr, vmin=None, vmax=None, cmap=None, format=None, If *format* is *None* and *fname* is a string, the output format is deduced from the extension of the filename. *arr*: - A 2D array. + An MxN (luminance), MxNx3 (RGB) or MxNx4 (RGBA) array. Keyword arguments: *vmin*/*vmax*: [ None | scalar ] *vmin* and *vmax* set the color scaling for the image by fixing the @@ -1240,7 +1240,7 @@ def imsave(fname, arr, vmin=None, vmax=None, cmap=None, format=None, from matplotlib.backends.backend_agg import FigureCanvasAgg as FigureCanvas from matplotlib.figure import Figure - figsize = [x / float(dpi) for x in arr.shape[::-1]] + figsize = [x / float(dpi) for x in (arr.shape[1], arr.shape[0])] fig = Figure(figsize=figsize, dpi=dpi, frameon=False) canvas = FigureCanvas(fig) im = fig.figimage(arr, cmap=cmap, vmin=vmin, vmax=vmax, origin=origin) -- 1.7.5.4