matplotlib grouped bar demo1.py

import numpy as np
import pandas as pd
import matplotlib.pyplot as plt

df = pd.read_csv('StudentsPerformance.csv')
subjects = ['math score', 'reading score', 'writing score']
dataset = df.groupby('gender')[subjects].mean()

indx = np.arange(len(subjects))
score_label = np.arange(0, 110, 10)
male_means = list(dataset.T['male'])
female_means = list(dataset.T['female'])

bar_width = 0.35

fig, ax = plt.subplots()
barMale = ax.bar(indx - bar_width/2, male_means, bar_width, label='Male means')
barFemale = ax.bar(indx + bar_width/2, female_means, bar_width, label='Female means')

# inserting x axis label
ax.set_xticks(indx)
ax.set_xticklabels(subjects)

# inserting y axis label
ax.set_yticks(score_label)
ax.set_yticklabels(score_label)

# inserting legend
ax.legend()

def insert_data_labels(bars):
	for bar in bars:
		bar_height = bar.get_height()
		ax.annotate('{0:.0f}'.format(bar.get_height()),
			xy=(bar.get_x() + bar.get_width() / 2, bar_height),
			xytext=(0, 3),
			textcoords='offset points',
			ha='center',
			va='bottom'
		)

insert_data_labels(barMale)
insert_data_labels(barFemale)

plt.show()