update

ecml_tools/data.py

@@ -90,6 +90,18 @@ def shape(self):
     def dates(self):
         return self.z.dates[:]  # Convert to numpy
 
+    @property
+    def latitudes(self):
+        return self.z.latitudes[:]
+
+    @property
+    def longitudes(self):
+        return self.z.longitudes[:]
+
+    @property
+    def statistics(self):
+        return self.z.statistics[:]
+
     @cached_property
     def resolution(self):
         return self.z.attrs["resolution"]
@@ -121,6 +133,14 @@ def check_compatibility(self, d1, d2):
                 f"Incompatible frequencies: {d1.frequency} and {d2.frequency} ({d1} {d2})"
             )
 
+    @property
+    def latitudes(self):
+        return self.datasets[0].latitudes
+
+    @property
+    def longitudes(self):
+        return self.datasets[0].longitudes
+
 
 class Concat(Combined):
     def __len__(self):
@@ -213,6 +233,14 @@ def shape(self):
     def dates(self):
         return self.dataset.dates[self.indices]
 
+    @property
+    def latitudes(self):
+        return self.dataset.latitudes
+
+    @property
+    def longitudes(self):
+        return self.dataset.longitudes
+
 
 def name_to_path(name):
     if isinstance(name, zarr.hierarchy.Group):

tests/test_code.py

@@ -16,7 +16,8 @@
 def _(date, var, k=0):
     d = date.year * 10000 + date.month * 100 + date.day
     v = ord(var) - ord("a") + 1
-    return d * 100 + v + k * 1_000_000_000
+    assert 0 <= k <= 9
+    return d * 100 + v + k / 10.0
 
 
 def create_zarr(
@@ -45,6 +46,7 @@ def create_zarr(
     root.dates = dates
     root.attrs["frequency"] = frequency
     root.attrs["resolution"] = 0
+    root.attrs["name_to_index"] = {k: i for i, k in enumerate(vars)}
 
     return root