tom/qubed
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

sort pressure levels

Browse Source
This commit is contained in:
Tom 2025-02-03 10:17:18 +00:00
parent 819c29d768
commit 8a2c5b341d
1 changed files with 10 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
fiab/extract.py
View File

@ -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)