Cristallina/cristallina_analysis_package
0
1
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 3 Wiki Activity

added custom value-to-color mapper

Browse Source
This commit is contained in:
steppke_a
2025-11-14 22:23:59 +01:00
parent aa51b3fc41
commit 6406fea7ed
1 changed files with 8 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
src/cristallina/plot.py
View File

@@ -2,6 +2,7 @@ import re
from collections import defaultdict from collections import defaultdict
import matplotlib import matplotlib
import matplotlib as mpl
from matplotlib import pyplot as plt from matplotlib import pyplot as plt
import warnings import warnings
@@ -13,7 +14,7 @@ import numpy as np
from tqdm import tqdm from tqdm import tqdm
from matplotlib import patches from matplotlib import patches
from pathlib import Path from pathlib import Path
import matplotlib as mpl
from sfdata import SFDataFiles, sfdatafile, SFScanInfo from sfdata import SFDataFiles, sfdatafile, SFScanInfo
import jungfrau_utils as ju import jungfrau_utils as ju
@@ -298,4 +299,9 @@ def line_plot_with_colorbar(xs,ys,colors, cmap=plt.cm.viridis,
ax.set_ylabel(ylabel) ax.set_ylabel(ylabel)
# add colorbar # add colorbar
fig.colorbar(s_m,ax=ax,ticks=colors,label=cbar_label,alpha=alpha) fig.colorbar(s_m,ax=ax,ticks=colors,label=cbar_label,alpha=alpha)
def get_normalized_colormap(vmin, vmax, cmap=mpl.cm.viridis):
norm = mpl.colors.Normalize(vmin=vmin, vmax=vmax)
return lambda x: cmap(norm(x))