Use scipy euclidean to measure distance.

dirac-institute · May 5, 2016 · 549e5bb · 549e5bb
1 parent 9e4d3ad
commit 549e5bb
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/code/createImage.py b/code/createImage.py
@@ -3,6 +3,7 @@
 import astropy.convolution as conv
 import matplotlib.pyplot as plt
 from astropy.io import fits
+from scipy.spatial.distance import euclidean
 
 class createImage(object):
 
@@ -377,7 +378,7 @@ def findLikelyTrajectories(self, psiArray, phiArray,
             testVel = topVel[rankings][objNum]
             keepVal = True
             for t0, vel in zip(keepT0, keepVel):
-                if ((np.sqrt(np.sum(np.power(testT0-t0,2))) < psfSigma) and (np.array_equal(testVel, vel))):
+                if ((euclidean(testT0, t0) < psfSigma) and (np.array_equal(testVel, vel))):
                     keepVal=False
             if keepVal == True:
                 keepT0[resultsSet] = testT0