[plotter] fix norm
This commit is contained in:
+10
-7
@@ -546,8 +546,9 @@ class Plotter(Aggregator, BaseProcessor):
|
|||||||
cmap="plasma",
|
cmap="plasma",
|
||||||
norm="log",
|
norm="log",
|
||||||
put_cbar=True,
|
put_cbar=True,
|
||||||
autoscale=True,
|
|
||||||
transform=None,
|
transform=None,
|
||||||
|
vmin=None,
|
||||||
|
vmax=None,
|
||||||
**kwargs,
|
**kwargs,
|
||||||
):
|
):
|
||||||
"""
|
"""
|
||||||
@@ -579,13 +580,15 @@ class Plotter(Aggregator, BaseProcessor):
|
|||||||
if transform is not None:
|
if transform is not None:
|
||||||
dmap = transform(dmap)
|
dmap = transform(dmap)
|
||||||
|
|
||||||
if norm == "log":
|
if vmin is None:
|
||||||
norm = mpl.colors.LogNorm()
|
vmin = np.min(dmap)
|
||||||
elif norm == "linear":
|
if vmax is None:
|
||||||
norm = mpl.colors.NoNorm()
|
vmax = np.max(dmap)
|
||||||
|
|
||||||
if autoscale and norm is not None:
|
if norm == "log":
|
||||||
norm.autoscale(dmap)
|
norm = mpl.colors.LogNorm(vmin=vmin, vmax=vmax)
|
||||||
|
elif norm == "linear":
|
||||||
|
norm = mpl.colors.Normalize(vmin=vmin, vmax=vmax)
|
||||||
|
|
||||||
im = plt.imshow(
|
im = plt.imshow(
|
||||||
dmap, extent=im_extent, origin="lower", norm=norm, cmap=cmap, **kwargs
|
dmap, extent=im_extent, origin="lower", norm=norm, cmap=cmap, **kwargs
|
||||||
|
|||||||
Reference in New Issue
Block a user