fixed issue in parser when trying to parse empty loop blocks (#71)

teamtomo · Oct 23, 2024 · a81abc3 · a81abc3
1 parent a2e9927
commit a81abc3
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/starfile/parser.py b/src/starfile/parser.py
@@ -116,9 +116,10 @@ def _parse_loop_block(self) -> pd.DataFrame:
             loop_data += '\n'
 
         # put string data into a dataframe
-        if loop_data == '\n':
+        if loop_data.startswith('\n'):
             n_cols = len(loop_column_names)
             df = pd.DataFrame(np.zeros(shape=(0, n_cols)))
+            df.columns = loop_column_names
         else:
             column_name_to_index = {col: idx for idx, col in enumerate(loop_column_names)}
             df = pd.read_csv(