diff --git a/modules/util/plot.py b/modules/util/plot.py index 831094dc906c88c00f5bf441c84f229f79ff79a6..c518ed8c4ba9f21fecfb659890916a2f9ad2f304 100644 --- a/modules/util/plot.py +++ b/modules/util/plot.py @@ -12,6 +12,8 @@ import xarray as xr import os import h5py from util.util import get_grid_values_all, get_cartopy_crs, homedir +import csv +from matplotlib.colors import LinearSegmentedColormap from util.setup import home_dir import mpl_scatter_density @@ -49,6 +51,19 @@ for i in range(2): cld_mask_cmap = ListedColormap(cld_mask_list, 'CldMask') +# Read colors from CSV file +colors_file = homedir + 'ICE.ct' +colors = [] +with open(colors_file, 'r') as csvfile: + csv_reader = csv.DictReader(csvfile) + for row in csv_reader: + colors.append((float(row['R']) / 255, float(row['G']) / 255, float(row['B']) / 255)) + +# Define a custom colormap +cmap_name = 'icing_colormap' +icing_cmap = LinearSegmentedColormap.from_list(cmap_name, colors, N=256) + + def plot_train_curve(rundir=None, title=None): last_epoch = 0 num_train_steps = 0