harmonic

1 files changed, 5 insertions, 2 deletions

diff --git a/scripts/average.py b/scripts/average.py
index 7aa7bfa2..dacd614b 100755
--- a/scripts/average.py
+++ b/scripts/average.py
@@ -12,9 +12,12 @@ for filename in sys.argv[1:-1]:
         m = np.load(mfile)
         for k in m:
             if k not in average:
-                average[k] = m[k] ** 1.0/n
+                average[k] = 1 / m[k]
             elif average[k].shape == m[k].shape:
-                average[k] *= m[k] ** 1.0/n
+                average[k] += 1 / m[k]
+
+for k in average:
+    average[k] = n / average[k]
 
 print "Saving to", sys.argv[-1]
 np.savez(sys.argv[-1], **average)