21 lines
659 B
Python
21 lines
659 B
Python
import matplotlib.pyplot as plt; plt.rcdefaults()
|
|
import numpy as np
|
|
import matplotlib.pyplot as plt
|
|
|
|
|
|
def plot_answers(answers, show=True, save=True, filename='distribution.png'):
|
|
answers = sorted(answers.items(), key=lambda kv : kv[1]['count'])
|
|
objects = [t[0] + ' (temp:{})'.format(round(t[1]['avgtemp'], 2)) 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')
|
|
if show:
|
|
plt.show()
|
|
if save:
|
|
plt.savefig('output/{}'.format(filename))
|