From 8a2c5b341dfec0eb3dedeb98511dca9ba328eb48 Mon Sep 17 00:00:00 2001 From: Tom Date: Mon, 3 Feb 2025 10:17:18 +0000 Subject: [PATCH] sort pressure levels --- fiab/extract.py | 11 ++++++++++- 1 file changed, 10 insertions(+), 1 deletion(-) diff --git a/fiab/extract.py b/fiab/extract.py index 3af796c..232747d 100644 --- a/fiab/extract.py +++ b/fiab/extract.py @@ -1,5 +1,6 @@ import json +from collections import defaultdict metadata = json.load(open("raw_anemoi_metadata.json")) @@ -15,8 +16,16 @@ pressure_level_variables = [v for v in variables if '_' in v] pressure_levels = sorted(set([v.split('_')[-1] for v in pressure_level_variables])) pressure_level_variables = sorted(set([v.split('_')[0] for v in pressure_level_variables])) +levels_for_variables = defaultdict(list) +for v in variables: + if "_" in v: + variable, level = v.split('_') + levels_for_variables[variable].append(level) + +print('Levels for variables:', levels_for_variables) + print('Pressure level variables:', pressure_level_variables) -print('Pressure levels:', pressure_levels) +print('Pressure levels:', sorted([int(p) for p in pressure_levels])) print('Surface variables:', surface_variables)