Add simple matplotlib bar plots per run

copycat/__init__.py

@@ -1 +1,2 @@
 from .copycat import Copycat, Reporter  # noqa
+from .plot    import plot_answers

copycat/plot.py

@@ -0,0 +1,18 @@
+import matplotlib.pyplot as plt; plt.rcdefaults()
+import numpy as np
+import matplotlib.pyplot as plt
+ 
+
+def plot_answers(answers):
+    answers = sorted(answers.items(), key=lambda kv : kv[1]['count'])
+    objects = [t[0] + ' (temp:{})'.format(t[1]['avgtemp']) for t in answers]
+    yvalues = [t[1]['count'] for t in answers]
+
+    y_pos = np.arange(len(objects))
+     
+    plt.bar(y_pos, yvalues, align='center', alpha=0.5)
+    plt.xticks(y_pos, objects)
+    plt.ylabel('Count')
+    plt.title('Answers')
+    plt.show()
+   

main.py

@@ -2,7 +2,7 @@
 import argparse
 import logging
 
-from copycat import Copycat, Reporter
+from copycat import Copycat, Reporter, plot_answers
 
 class SimpleReporter(Reporter):
     def report_answer(self, answer):
@@ -27,5 +27,7 @@ def main():
     for answer, d in sorted(iter(answers.items()), key=lambda kv: kv[1]['avgtemp']):
         print('%s: %d (avg time %.1f, avg temp %.1f)' % (answer, d['count'], d['avgtime'], d['avgtemp']))
 
+    plot_answers(answers)
+
 if __name__ == '__main__':
     main()