Merge pull request #247 from TGSAI/fix/nonzero_calculation

Update logic for finding first non-zero sample index in SEG-Y data
TGSAI · Jul 24, 2023 · 187ca3f · 187ca3f
2 parents 5216791 + 9c7c767
commit 187ca3f
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/mdio/segy/_workers.py b/src/mdio/segy/_workers.py
@@ -207,7 +207,10 @@ def trace_worker(
         value=tmp_metadata,
     )
 
-    nonzero_z = tmp_data.sum(axis=tuple(range(n_dim - 1))).nonzero()
+    # Find first non-zero index in sample (last) dimension
+    non_sample_axes = tuple(range(n_dim - 1))
+    nonzero_z = np.where(np.any(tmp_data != 0, axis=non_sample_axes))
+
     if len(nonzero_z[0]) == 0:
         return