Merge pull request #333 from ErlendHaa/performance-writing-lines

Write each trace only on time in HeaderLine
equinor · Nov 29, 2018 · 7a9954f · 7a9954f
2 parents 7dc9e3b + 3b5b943
commit 7a9954f
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/python/segyio/line.py b/python/segyio/line.py
@@ -427,7 +427,8 @@ def __getitem__(self, index):
 
         else:
             step = self.stride * len(self.offsets)
-            return self.header[start::step]
+            stop = start + step * self.length
+            return self.header[start:stop:step]
 
         def gen():
             irange, orange = self.ranges(index, offset)
@@ -496,7 +497,8 @@ def __setitem__(self, index, val):
                 pass
 
             step = self.stride * len(self.offsets)
-            self.header[start::step] = val
+            stop  = start + step * self.length
+            self.header[start:stop:step] = val
             return
 
         irange, orange = self.ranges(index, offset)