diff options
| -rw-r--r-- | math_utils.py | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/math_utils.py b/math_utils.py index dd26cb6..64cd9fb 100644 --- a/math_utils.py +++ b/math_utils.py @@ -96,7 +96,7 @@ class NumericPopulation(object): variance += (n - mean) ** 2 for n in self.highers: variance += (n - mean) ** 2 - count = len(self.lowers) + len(self.highers) - 1 + count = len(self.lowers) + len(self.highers) return math.sqrt(variance) / count def _create_sorted_copy_if_needed(self, count: int): @@ -120,7 +120,8 @@ class NumericPopulation(object): self._create_sorted_copy_if_needed(count) assert self.sorted_copy index = round(count * (n / 100.0)) - assert 0 <= index < count + index = max(0, index) + index = min(count - 1, index) return self.sorted_copy[index] |