snapshot...

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