do not overwrite input, use specialized function
This commit is contained in:
parent
39dacbc972
commit
c0a97a93fe
|
@ -305,11 +305,11 @@ class Colormap(mpl.colors.ListedColormap):
|
|||
if abs(delta := r-l) * 1e8 <= (avg := 0.5*abs(r+l)): # delta is similar to numerical noise
|
||||
l,r = (l-0.5*avg*np.sign(delta),r+0.5*avg*np.sign(delta)) # extend range to have actual data centered within
|
||||
|
||||
field[np.isnan(field)] = (l+r)/2
|
||||
field_ = np.nan_to_num(field, nan=(l+r)/2, posinf=r, neginf=l)
|
||||
|
||||
return Image.fromarray(
|
||||
(np.dstack((
|
||||
self.colors[np.round(np.clip((field-l)/(r-l),0.0,1.0)*(self.N-1)).astype(np.uint16),:3],
|
||||
self.colors[np.round(np.clip((field_-l)/(r-l),0.0,1.0)*(self.N-1)).astype(np.uint16),:3],
|
||||
mask.astype(float)
|
||||
)
|
||||
)*255
|
||||
|
|
Loading…
Reference in New Issue