Commit9512d32

committed

ENH: add a copy method to colormaps

1 parentbd7df78 commit9512d32Copy full SHA for 9512d32

File tree

+17

-1

lines changed

+17

-1

lines changed

Lines changed: 7 additions & 1 deletion

Original file line number	Diff line number	Diff line change
`@@ -561,7 +561,7 @@ def _warn_if_global_cmap_modified(cmap):`
`561`	`561`	`"colormap. In future versions, you will not be able to "`
`562`	`562`	`"modify a registered colormap in-place. To remove this "`
`563`	`563`	`"warning, you can make a copy of the colormap first. "`
`564`		`-f'cmap =copy.copy(mpl.cm.get_cmap("{cmap.name}"))'`
	`564`	`+f'cmap = mpl.cm.get_cmap("{cmap.name}").copy()'`
`565`	`565`	`)`
`566`	`566`
`567`	`567`
`@@ -845,6 +845,12 @@ def color_block(color):`
`845`	`845`	`f'over{color_block(self.get_over())}'`
`846`	`846`	`'</div>')`
`847`	`847`
	`848`	`+defcopy(self):`
	`849`	`+"""`
	`850`	`+ Return a copy of the colormap.`
	`851`	`+ """`
	`852`	`+returnself.__copy__()`
	`853`	`+`
`848`	`854`
`849`	`855`	`classLinearSegmentedColormap(Colormap):`
`850`	`856`	`"""`

Lines changed: 10 additions & 0 deletions

Original file line number	Diff line number	Diff line change
`@@ -150,6 +150,16 @@ def test_colormap_copy():`
`150`	`150`	`withnp.errstate(invalid='ignore'):`
`151`	`151`	`ret2=copied_cmap([-1,0,.5,1,np.nan,np.inf])`
`152`	`152`	`assert_array_equal(ret1,ret2)`
	`153`	`+# again with the .copy method:`
	`154`	`+cmap=plt.cm.Reds`
	`155`	`+copied_cmap=cmap.copy()`
	`156`	`+withnp.errstate(invalid='ignore'):`
	`157`	`+ret1=copied_cmap([-1,0,.5,1,np.nan,np.inf])`
	`158`	`+cmap2=copy.copy(copied_cmap)`
	`159`	`+cmap2.set_bad('g')`
	`160`	`+withnp.errstate(invalid='ignore'):`
	`161`	`+ret2=copied_cmap([-1,0,.5,1,np.nan,np.inf])`
	`162`	`+assert_array_equal(ret1,ret2)`
`153`	`163`
`154`	`164`
`155`	`165`	`deftest_colormap_endian():`

Comments

(0)