side_by_side_bar.py

import codecademylib
from matplotlib import pyplot as plt

unit_topics = ['Limits', 'Derivatives', 'Integrals', 'Diff Eq', 'Applications']
middle_school_a = [80, 85, 84, 83, 86]
middle_school_b = [73, 78, 77, 82, 86]

def create_x(t, w, n, d):
    return [t*x + w*n for x in range(d)]
school_a_x = [0.8, 2.8, 4.8, 6.8, 8.8]
school_b_x = [1.6, 3.6, 5.6, 7.6, 9.6]
# Make your chart here
school_a_x=[2*t +0.8 for t in range(len(unit_topics))]
school_b_x=[2*t+0.8*2 for t in range(len(unit_topics))]
plt.figure(figsize=(10,8))
ax=plt.subplot()
plt.bar(school_a_x,middle_school_a)
plt.bar(school_b_x,middle_school_b)
middle_x=[(a+b)/2 for a,b in zip(school_a_x,school_b_x)]
ax.set_xticks(middle_x)
ax.set_xticklabels(unit_topics)
plt.legend(['Middle School A','Middle School B'])
plt.title('Test Averages on Different Units')
plt.xlabel('Unit')
plt.ylabel('Test Average')
plt.show()
plt.savefig('my_side_by_side.png')