Average_employee.py

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

header = ['monthly income','job role','marital status','overtime', 'stock option level', 'age']
dataset = [(100, 8.9, 32.0, 28.3, 42.9, 100),(0, 3.5,45.8,71.7,40.5,0), (0,17.6,22.2,0,10.7,0), (0,6.9,0,0,5.8,0), (0,9.9,0,0,0,0), (0,5.4,0,0,0,0), (0,19.9,0,0,0,0), (0,22.2,0,0,0,0), (0,5.6,0,0,0,0)]

fig = matplotlib.pyplot.gcf()
fig.set_size_inches(18.5, 10.5)
configs = dataset[0]
N = 6
ind = np.arange(N)
width = 0.7

p1 = plt.bar(ind, dataset[0], width, color='#7030A2')
p2 = plt.bar(ind, dataset[1], width, bottom=dataset[0], color='#FD5854')
p3 = plt.bar(ind, dataset[2], width, 
             bottom=np.array(dataset[0])+np.array(dataset[1]), color='#EA5FC2')
p4 = plt.bar(ind, dataset[3], width,
             bottom=np.array(dataset[0])+np.array(dataset[1])+np.array(dataset[2]),
             color='#CE2BA9')
p5 = plt.bar(ind, dataset[4], width,
             bottom=np.array(dataset[0])+np.array(dataset[1])+np.array(dataset[2])+np.array(dataset[3]),
             color='#F6AEF0')
p6 = plt.bar(ind, dataset[5], width,
             bottom=np.array(dataset[0])+np.array(dataset[1])+np.array(dataset[2])+np.array(dataset[3])+np.array(dataset[4]), 
             color='#9052BC')
p7 = plt.bar(ind, dataset[6], width,
             bottom=np.array(dataset[0])+np.array(dataset[1])+np.array(dataset[2])+np.array(dataset[3])+np.array(dataset[4])+np.array(dataset[5]),
             color='#BF9BB5')
p8 = plt.bar(ind, dataset[7], width,
             bottom=np.array(dataset[0])+np.array(dataset[1])+np.array(dataset[2])+np.array(dataset[3])+np.array(dataset[4])+np.array(dataset[5])+np.array(dataset[6]),
             color='#822240')
p9 = plt.bar(ind, dataset[8], width,
             bottom=np.array(dataset[0])+np.array(dataset[1])+np.array(dataset[2])+np.array(dataset[3])+np.array(dataset[4])+np.array(dataset[5])+np.array(dataset[6])+np.array(dataset[7]),
             color='#AF5D71')
             
plt.title('Average employee', fontsize=30)
plt.ylim([0,110])
plt.yticks(ind,"")
plt.xticks(ind, header, fontsize=15, rotation=0)

plt.text(-0.21,50, '6503', color='white', fontsize=25, fontweight='bold')
plt.text(0.7, 5, 'Healthcare', color='white', fontsize=15, fontweight='bold')
plt.text(0.89, 2, 'Rep', color='white', fontsize=15, fontweight='bold')
plt.text(0.9, 9.5, 'HR', color='black', fontsize=15, fontweight='bold')
plt.text(0.73, 22, 'Laboratory', color='black', fontsize=15, fontweight='bold')
plt.text(0.73, 18, 'Technician', color='black', fontsize=15, fontweight='bold')
plt.text(0.78, 32, 'Manager', color='black', fontsize=15, fontweight='bold')
plt.text(0.68, 43, 'Manufacturing', color='black', fontsize=13, fontweight='bold')
plt.text(0.8, 40, 'Director', color='black', fontsize=13, fontweight='bold')
plt.text(0.78, 50, 'Research', color='black', fontsize=13, fontweight='bold')
plt.text(0.8, 47.5, 'Director', color='black', fontsize=13, fontweight='bold')
plt.text(0.76, 63, 'Research', color='black', fontsize=15, fontweight='bold')
plt.text(0.78, 59, 'Scientist', color='black', fontsize=15, fontweight='bold')
plt.text(0.85, 85, 'Sales', color='white', fontsize=15, fontweight='bold')
plt.text(0.75, 80, 'Executive', color='white', fontsize=15, fontweight='bold')
plt.text(0.75, 96, 'Sales Rep', color='black', fontsize=15, fontweight='bold')
plt.text(1.73, 16, 'Single', color='white', fontsize=25, fontweight='bold')
plt.text(1.7, 55, 'Married', color='black', fontsize=23, fontweight='bold')
plt.text(1.7, 87, 'Divorced', color='black', fontsize=20, fontweight='bold')
plt.text(2.85, 14, 'Yes', color='white', fontsize=25, fontweight='bold')
plt.text(2.86, 63, 'No', color='black', fontsize=30, fontweight='bold')
plt.text(3.7, 20, 'Level 0', color='white', fontsize=23, fontweight='bold')
plt.text(3.7, 61, 'Level 1', color='black', fontsize=23, fontweight='bold')
plt.text(3.75, 87, 'Level 2', color='black', fontsize=20, fontweight='bold')
plt.text(3.82, 96, 'Level 3', color='black', fontsize=15, fontweight='bold')
plt.text(4.87,50, '37', color='white', fontsize=30, fontweight='bold')

plt.show()