Merge pull request #1920 from jamesmkrieger/catdcd_first

add first to catdcd goto
prody · Jul 29, 2024 · 555f2f9 · 555f2f9
2 parents 098e5ec + e66eccf
commit 555f2f9
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/prody/apps/prody_apps/prody_catdcd.py b/prody/apps/prody_apps/prody_catdcd.py
@@ -80,8 +80,9 @@ def prody_catdcd(*dcd, **kwargs):
     out = prody.DCDFile(output, 'w')
     count = 0
     stride = kwargs.get('stride', 1)
-    goto = stride != 1
-    slc = slice(kwargs.get('first', 0), kwargs.get('last', -1),
+    first = kwargs.get('first', 0)
+    goto = stride != 1 or first != 0
+    slc = slice(first, kwargs.get('last', -1),
                 stride).indices(len(traj)+1)
     for i in range(*slc):
         if goto: