From d3cb761abb18878cd03e931c448630ab4818b040 Mon Sep 17 00:00:00 2001
From: Daniel Toyra <dtoyra@star.sr.bham.ac.uk>
Date: Fri, 24 Jun 2016 00:56:19 +0100
Subject: [PATCH] Changed from pylab to pyplot for plotting absorption map
---
pykat/optics/maps.py | 19 ++++++++++---------
1 file changed, 10 insertions(+), 9 deletions(-)
diff --git a/pykat/optics/maps.py b/pykat/optics/maps.py
index d444d9c..ba0970c 100644
--- a/pykat/optics/maps.py
+++ b/pykat/optics/maps.py
@@ -1604,7 +1604,8 @@ class mergedmap:
def plot(self, mode="absorption", show=True, clabel=None, xlim=None, ylim=None, wavelength=1064e-9, isBlock=False):
- import pylab
+ import matplotlib
+ import matplotlib.pyplot as plt
if xlim is not None:
_x = np.logical_and(self.x<=max(xlim)/100.0, self.x>=min(xlim)/100.0)
@@ -1639,15 +1640,15 @@ class mergedmap:
xrange = 100*self.x
yrange = 100*self.y
- fig = pylab.figure()
- axes = pylab.pcolormesh(xrange, yrange, data, vmin=zmin, vmax=zmax)
- pylab.xlabel('x [cm]')
- pylab.ylabel('y [cm]')
+ fig = plt.figure()
+ axes = plt.pcolormesh(xrange, yrange, data, vmin=zmin, vmax=zmax)
+ plt.xlabel('x [cm]')
+ plt.ylabel('y [cm]')
- if xlim is not None: pylab.xlim(xlim)
- if ylim is not None: pylab.ylim(ylim)
+ if xlim is not None: plt.xlim(xlim)
+ if ylim is not None: plt.ylim(ylim)
- pylab.title('Merged map {0}, mode {1}'.format(self.name, mode))
+ plt.title('Merged map {0}, mode {1}'.format(self.name, mode))
cbar = fig.colorbar(axes)
cbar.set_clim(zmin, zmax)
@@ -1656,7 +1657,7 @@ class mergedmap:
cbar.set_label(clabel)
if show:
- pylab.show(block=isBlock)
+ plt.show(block=isBlock)
return fig
--
GitLab