add a Cluster.center() method

craigds · Oct 25, 2011 · 1da7c09 · 1da7c09
1 parent 45163b2
commit 1da7c09
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/pygvm/__init__.py b/pygvm/__init__.py
@@ -31,7 +31,10 @@ def __init__(self, clusters):
         self.pairs = []
 
     def center(self):
-        return self.m1 / self.mass
+        if self.mass:
+            return [coord / self.mass for coord in self.m1]
+        else:
+            return self.m1[:]
 
     def clear(self):
         """