How to write text above the bars on a bar plot (Python)?
plt.bar()
returns a list of rectangles that can be used to position suitable text above each of the bars as follows:
import matplotlib.pyplot as plt
import numpy as np
errorRateListOfFast = ['9.09', '9.09', '9.38', '9.40', '7.89', '8.02', '10.00']
errorRateListOfSlow = ['10.00', '13.04', '14.29', '12.50', '14.29', '14.53', '11.11']
# Convert to floats
errorRateListOfFast = [float(x) for x in errorRateListOfFast]
errorRateListOfSlow = [float(x) for x in errorRateListOfSlow]
opacity = 0.4
bar_width = 0.35
plt.xlabel('Tasks')
plt.ylabel('Error Rate')
plt.xticks(range(len(errorRateListOfFast)),('[10-20)', '[20-30)', '[30-50)', '[50-70)','[70-90)', '[90-120)', ' [120 < )'), rotation=30)
bar1 = plt.bar(np.arange(len(errorRateListOfFast)) + bar_width, errorRateListOfFast, bar_width, align='center', alpha=opacity, color='b', label='Fast <= 6 sec.')
bar2 = plt.bar(range(len(errorRateListOfSlow)), errorRateListOfSlow, bar_width, align='center', alpha=opacity, color='r', label='Slower > 6 sec.')
# Add counts above the two bar graphs
for rect in bar1 + bar2:
height = rect.get_height()
plt.text(rect.get_x() + rect.get_width() / 2.0, height, f'{height:.0f}', ha='center', va='bottom')
plt.legend()
plt.tight_layout()
plt.show()
Giving you:
ha='center'
and va='bottom'
refer to how the text is aligned in relation to the x
and y
co-ordinates, i.e. horizontal and vertical alignment.
Today you can use this simple snippet:
ax = plt.bar(...)
ax.bar_label(ax.containers[0])
Check out the following link, it may help:
http://matplotlib.org/examples/api/barchart_demo.html