Dynamically calc index

microsoft · Oct 4, 2023 · e66a1c2 · e66a1c2
1 parent 5c5dfaf
commit e66a1c2
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/qcodes/dataset/data_set.py b/qcodes/dataset/data_set.py
@@ -1488,10 +1488,13 @@ def _export_as_netcdf(self, path: Path, file_name: str) -> Path:
                         "temp_dir": temp_dir,
                     },
                 )
-                for i in trange(len(self), desc="Writing individual files"):
+                num_files = len(self)
+                num_digits = len(str(num_files))
+                file_name_template = f"ds_{{:0{num_digits}d}}.nc"
+                for i in trange(num_files, desc="Writing individual files"):
                     xarray_to_h5netcdf_with_complex_numbers(
                         self.to_xarray_dataset(start=i + 1, end=i + 1),
-                        temp_path / f"ds_{i:03d}.nc",
+                        temp_path / file_name_template.format(i),
                     )
                 files = tuple(temp_path.glob("*.nc"))
                 data = xr.open_mfdataset(files)