fix(ParticleTrackFile.write_shapefile): check for "k" even if "i", "…

…j are not present (modflowpy#2294) * fix for "k", "node" output present with mp7 pathline and timeseries
mjreno · Aug 17, 2024 · f86881d · f86881d
1 parent 8656e4e
commit f86881d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/flopy/utils/particletrackfile.py b/flopy/utils/particletrackfile.py
@@ -291,11 +291,11 @@ def write_shapefile(
                 if "particlegroup" in names:
                     t.append(ra.particlegroup[0])
                 t.append(ra.time.max())
+                if "k" in names:
+                    t.append(ra.k[loc_inds])
                 if "node" in names:
                     t.append(ra.node[loc_inds])
                 else:
-                    if "k" in names:
-                        t.append(ra.k[loc_inds])
                     if "i" in names:
                         t.append(ra.i[loc_inds])
                     if "j" in names: