Stem plot in matplotlib?
There is a stem plot (a.k.a. lollipop plot) in the matplotlib as well.
Below you can find an example from the docs.
import matplotlib.pyplot as plt
import numpy as np
# returns 10 evenly spaced samples from 0.1 to 2*PI
x = np.linspace(0.1, 2 * np.pi, 10)
markerline, stemlines, baseline = plt.stem(x, np.cos(x), '-.')
# setting property of baseline with color red and linewidth 2
plt.setp(baseline, color='r', linewidth=2)
plt.show()
If you want to adjust the y-position of the baseline, you can use the bottom
parameter.
(adapted) Example:
import matplotlib.pyplot as plt
import numpy as np
# returns 10 evenly spaced samples from 0.1 to 2*PI
x = np.linspace(0.1, 2 * np.pi, 10)
plt.stem(x, np.cos(x), '-.', bottom=-2)
plt.show()