Logo Questions Linux Laravel Mysql Ubuntu Git Menu
 

Matplotlib Plot Lines Above Each Bar

I would like to plot a horizontal line above each bar in this chart. The y-axis location of each bar depends on the variable 'target.' I want to use axhline, if possible, or Line2D because I need to be able to modify the line style, color, length, and width.

import matplotlib.pyplot as plt
plt.rcdefaults()
import numpy as np
import matplotlib.pyplot as plt
%matplotlib inline


# Example data
people = ('Tom', 'Dick', 'Harry', 'Slim', 'Jim')

#Here are the targets that I want to use 
#to plot horizontal lines above each bar...
targets = (6,6,8,6,9)

ind = np.arange(len(people))
performance = 3 + 10 * np.random.rand(len(people))
error = np.random.rand(len(people))

plt.bar(ind, performance, align='center')
plt.xticks(ind, people)

plt.show()

Thanks in advance!

like image 403
Dance Party2 Avatar asked Nov 30 '15 20:11

Dance Party2


1 Answers

You need to provide the x_start and x_end to .hlines(). They can be numpy.arrays, in which case each element determines the start/end point of each line segment:

import matplotlib.pyplot as plt
plt.rcdefaults()
import numpy as np
import matplotlib.pyplot as plt
%matplotlib inline


# Example data
people = ('Tom', 'Dick', 'Harry', 'Slim', 'Jim')

#Here are the targets that I want to use 
#to plot horizontal lines above each bar...
targets = (6,6,8,6,9)

ind = np.arange(len(people))
performance = 3 + 10 * np.random.rand(len(people))
error = np.random.rand(len(people))

Bars = plt.bar(ind, performance, align='center')
_ = plt.xticks(ind, people)

#if you want your hlines to align with the bars.
#i.e. start and end at the same x coordinates:

x_start = np.array([plt.getp(item, 'x') for item in Bars])
x_end   = x_start+[plt.getp(item, 'width') for item in Bars]

plt.hlines(targets, x_start, x_end)

enter image description here

like image 180
CT Zhu Avatar answered Oct 22 '22 00:10

CT Zhu