|
63
|
1 #!/usr/bin/env python
|
|
|
2 # -*- coding: utf-8 -*-
|
|
|
3
|
|
|
4 """
|
|
68
|
5 Histograms
|
|
63
|
6
|
|
|
7 Unicode is awesome; see http://www.alanwood.net/unicode/block_elements.html
|
|
|
8 """
|
|
|
9
|
|
|
10 blocks = """
|
|
|
11 █
|
|
|
12 ▉
|
|
|
13 ▊
|
|
|
14 ▋
|
|
|
15 ▌
|
|
|
16 ▍
|
|
|
17 ▎
|
|
|
18 ▏
|
|
|
19 """
|
|
|
20
|
|
|
21 # imports
|
|
|
22 import argparse
|
|
|
23 import os
|
|
|
24 import sys
|
|
|
25 import time
|
|
|
26 from .read import CSVParser
|
|
64
|
27 from collections import OrderedDict
|
|
63
|
28
|
|
|
29 # module globals
|
|
64
|
30 __all__ = ['Histogram', 'HistogramParser', 'main']
|
|
|
31
|
|
|
32 class Histogram(object):
|
|
|
33 """historgram"""
|
|
63
|
34
|
|
64
|
35 def __init__(self, bins):
|
|
|
36 bins = sorted(bin)
|
|
|
37 assert len(bins) > 1
|
|
68
|
38 self.data = self.OrderedDict(zip(bins[:-1],
|
|
|
39 bins[1:]))
|
|
64
|
40
|
|
70
|
41 def add(self, *values):
|
|
|
42 """add values to the histogram"""
|
|
71
|
43 for value in values:
|
|
|
44 raise NotImplementedError('TODO') # -> record TODO items
|
|
70
|
45
|
|
65
|
46 def __iadd__(self, value):
|
|
71
|
47 self.add(value)
|
|
65
|
48 return self
|
|
|
49
|
|
71
|
50 def __call__(self, *values):
|
|
|
51 """
|
|
|
52 add values to the histogram and return
|
|
|
53 OrderedDict of counts
|
|
|
54 """
|
|
|
55 self.add(*values)
|
|
|
56
|
|
64
|
57 class HistogramParser(CSVParser):
|
|
63
|
58 """histogram CLI option parser"""
|
|
|
59
|
|
|
60 def __init__(self, **kwargs):
|
|
|
61 kwargs.setdefault('formatter_class', argparse.RawTextHelpFormatter)
|
|
|
62 kwargs.setdefault('description', __doc__)
|
|
69
|
63 CSVParser.__init__(self, **kwargs)
|
|
63
|
64 self.add_argument('-n', '--bins', dest='n_bins', type=int,
|
|
|
65 help="number of bins")
|
|
|
66 self.options = None
|
|
|
67
|
|
|
68
|
|
|
69 def main(args=sys.argv[1:]):
|
|
|
70 """CLI"""
|
|
|
71
|
|
|
72 # parse command line options
|
|
|
73 parser = HistogramParser()
|
|
|
74 options = parser.parse_args(args)
|
|
|
75
|
|
|
76
|
|
|
77 if __name__ == '__main__':
|
|
|
78 main()
|
|
|
79
|
|
|
80
|
