Flip image without modifying attribute when writing PNG

naoyak · naoyak · commit abcdf28306f3 · 2017-01-03T14:15:47.000-08:00
diff --git a/lib/matplotlib/image.py b/lib/matplotlib/image.py
@@ -521,9 +521,8 @@ def contains(self, mouseevent):
 
     def write_png(self, fname):
         """Write the image to png file with fname"""
-        if self.origin == 'upper':
-            self._A = self._A[::-1]
-        im = self.to_rgba(self._A, bytes=True, norm=True)
+        im = self.to_rgba(self._A[::-1] if self.origin == 'upper' else self._A,
+                          bytes=True, norm=True)
         _png.write_png(im, fname)
 
     def set_data(self, A):
