arithmetic

1 files changed, 2 insertions, 5 deletions

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