Added __copy__ to Colormap

vidursatija · web-flow · commit 3ed55f480ccc · 2017-03-17T10:01:36.000+05:30
diff --git a/lib/matplotlib/colors.py b/lib/matplotlib/colors.py
@@ -529,7 +529,18 @@ def __call__(self, X, alpha=None, bytes=False):
         if vtype == 'scalar':
             rgba = tuple(rgba[0, :])
         return rgba
-
+    
+    def __copy__(self):
+        """Create new object with the same class and update attributes
+           If object is initialized, copy the elements of _lut list
+        """
+        cls = self.__class__
+    	newCMapObj = cls.__new__(cls)
+    	newCMapObj.__dict__.update(self.__dict__)
+        if self._isinit:
+            newCMapObj._lut = np.copy(self._lut)
+        return newCMapObj
+    
     def set_bad(self, color='k', alpha=None):
         """Set color to be used for masked values.
         """
