@@ -82,10 +82,10 @@ def write(
8282 if dtype == object :
8383 dtype = "S30"
8484
85- self .store_datasets [obj_type ][col_name ] = self .store_groups [
85+ self .store_datasets [obj_type ][self . column_name ( col_name ) ] = self .store_groups [
8686 obj_type
8787 ].create_dataset (
88- str (col_name ),
88+ self . column_name (col_name ),
8989 shape = (self .column_length ,),
9090 maxshape = (None ,),
9191 chunks = True ,
@@ -109,10 +109,10 @@ def write(
109109 si = int (self .step / self .chunkRows ) * self .chunkRows
110110 ei = si + self .chunkRows
111111 for col_name in powerflow_output [obj_type ].keys ():
112- r = self .store_datasets [obj_type ][col_name ].shape [0 ]
112+ r = self .store_datasets [obj_type ][self . column_name ( col_name ) ].shape [0 ]
113113 if ei >= r :
114- self .store_datasets [obj_type ][col_name ].resize ((ei ,))
115- self .store_datasets [obj_type ][col_name ][si :ei ] = self .dfs [
114+ self .store_datasets [obj_type ][self . column_name ( col_name ) ].resize ((ei ,))
115+ self .store_datasets [obj_type ][self . column_name ( col_name ) ][si :ei ] = self .dfs [
116116 obj_type
117117 ][col_name ]
118118 self .dfs [obj_type ] = None
@@ -125,6 +125,9 @@ def write(
125125 self .store .flush ()
126126 self .step += 1
127127
128+ def column_name (self , column_name :str )-> str :
129+ return str (column_name ).replace (" " , "" )
130+
128131 def close_store (self ):
129132 try :
130133 k = list (self .dfs .keys ())[0 ]
@@ -133,7 +136,7 @@ def close_store(self):
133136 if length > 0 :
134137 for obj_type in self .dfs .keys ():
135138 for col_name in self .dfs [k ].columns :
136- self .store_datasets [obj_type ][col_name ][
139+ self .store_datasets [obj_type ][self . column_name ( col_name ) ][
137140 self .column_length - length :
138141 ] = self .dfs [obj_type ][col_name ]
139142
0 commit comments