Histogram.py 473 B

123456789101112131415161718
  1. class Histogram:
  2. def __init__(self, bins, min, max):
  3. self.bins = [0] * bins
  4. self.min = min
  5. self.max = max
  6. def add(self, number):
  7. bin_index = self.get_bin_index(number)
  8. self.bins[bin_index] += 1
  9. def remove(self, number):
  10. bin_index = self.get_bin_index(number)
  11. self.bins[bin_index] -= 1
  12. def get_bin_index(self, number):
  13. return (number - self.min) / ((self.max - self.min) / len(self.bins))