loadnpy: do not slice the same object twice (#296)

storm-fsv-cvut · Nov 22, 2023 · 484e050 · 484e050
1 parent 113aba3
commit 484e050
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/smoderp2d/tools/save_load_data_nopickle.py b/smoderp2d/tools/save_load_data_nopickle.py
@@ -87,8 +87,8 @@ def loadunicode(self):
         return n[0]
 
     def loadnpy(self):
-
-        n = len(self.lines[2:])
+        rows = self.lines[2:]
+        n = len(rows)
         m = len(self.lines[2].split(';'))
         type_ = self.lines[1]
         arr = np.zeros([n, m], float)
@@ -98,7 +98,7 @@ def loadnpy(self):
         if 'float' in type_:
             self.npyel = self.__float
 
-        for i, line in enumerate(self.lines[2:]):
+        for i, line in enumerate(rows):
             for j, el in enumerate(line.split(';')):
                 arr[i, j] = self.npyel(el)