X-Git-Url: https://wannabe.guru.org/gitweb/?a=blobdiff_plain;f=math_utils.py;h=dec34f049aa0382823768c53efc2060ae8c69409;hb=068d71327bf6ec8618cd70a6d3fce5d075503cae;hp=3a672da1caf406d3a842ae46923a88063743dbbd;hpb=699219341e92365b7aa2f84df0c141bc4f0aa238;p=python_utils.git

diff --git a/math_utils.py b/math_utils.py
index 3a672da..dec34f0 100644
--- a/math_utils.py
+++ b/math_utils.py
@@ -1,11 +1,16 @@
 #!/usr/bin/env python3
 
+# © Copyright 2021-2022, Scott Gasch
+
 """Mathematical helpers."""
 
+import collections
 import functools
 import math
 from heapq import heappop, heappush
-from typing import List, Optional
+from typing import Dict, List, Optional, Tuple
+
+import dict_utils
 
 
 class NumericPopulation(object):
@@ -70,6 +75,16 @@ class NumericPopulation(object):
         count = len(self.lowers) + len(self.highers)
         return self.aggregate / count
 
+    def get_mode(self) -> Tuple[float, int]:
+        """Returns the mode (most common member)."""
+
+        count: Dict[float, int] = collections.defaultdict(int)
+        for n in self.lowers:
+            count[-n] += 1
+        for n in self.highers:
+            count[n] += 1
+        return dict_utils.item_with_max_value(count)
+
     def get_stdev(self) -> float:
         """Returns the stdev so far in O(n) time."""