Add geometric mean calculation to analysis script

sharkdp · Jul 11, 2024 · ba31fa9 · ba31fa9
1 parent 824aafb
commit ba31fa9
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/scripts/advanced_statistics.py b/scripts/advanced_statistics.py
@@ -3,6 +3,7 @@
 import argparse
 import json
 import numpy as np
+from scipy.stats import gmean
 
 parser = argparse.ArgumentParser()
 parser.add_argument("file", help="JSON file with benchmark results")
@@ -25,6 +26,7 @@
     print("Command '{}'".format(command))
     print("  runs:   {:8d}".format(len(ts)))
     print("  mean:   {:8.3f} s".format(np.mean(ts)))
+    print("  geomean:{:8.3f} s".format(gmean(ts)))
     print("  stddev: {:8.3f} s".format(np.std(ts, ddof=1)))
     print("  median: {:8.3f} s".format(np.median(ts)))
     print("  min:    {:8.3f} s".format(np.min(ts)))