io: Add compression to series attributes in netcdf exporter

openego · Dec 18, 2017 · 352598b · 352598b
1 parent a54b58c
commit 352598b
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/pypsa/io.py b/pypsa/io.py
@@ -220,8 +220,9 @@ def get_series(self, list_name):
                     yield attr[len(t):], df
 
     class ExporterNetCDF(Exporter):
-        def __init__(self, path):
+        def __init__(self, path, least_significant_digit=3):
             self.path = path
+            self.least_significant_digit = least_significant_digit
             self.ds = xr.Dataset()
 
         def save_attributes(self, attrs):
@@ -242,6 +243,11 @@ def save_series(self, list_name, attr, df):
             df.index.name = 'snapshots'
             df.columns.name = list_name + '_t_' + attr + '_i'
             self.ds[list_name + '_t_' + attr] = df
+            if self.least_significant_digit is not None:
+                self.ds.encoding.upate({
+                    'zlib': True,
+                    'least_significant_digit': self.least_significant_digit
+                })
 
         def finish(self):
             if self.path is not None: