Changed from pylab to pyplot for plotting absorption map

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