Update vignette_space.md

Suggesting geographic distance to use only the first two entries of locs, in case the third entries are not all equal to each other.
tskit-dev · Nov 8, 2023 · 263ca53 · 263ca53
1 parent f991488
commit 263ca53
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/docs/vignette_space.md b/docs/vignette_space.md
@@ -357,8 +357,8 @@ geog_dist = np.repeat(0.0, len(pairs))
 locs = ts.individuals_location
 for k, (i, j) in enumerate(pairs):
   geog_dist[k] = np.sqrt(np.sum(
-                    (locs[ind_ids[i], :]
-                     - locs[ind_ids[j], :])**2
+                    (locs[ind_ids[i], :2]
+                     - locs[ind_ids[j], :2])**2
                  ))
 ```