Skip to content Skip to sidebar Skip to footer

Label Areas In Python Matplotlib Stackplot

I would like to generate labels inside the areas of a matplotlib stackplot. I would settle for labeling a line used to bound the area. Consider the example: import numpy as np from

Solution 1:

Ah, heuristics. Something like this?:

import numpy as np
from matplotlib import pyplot as plt

length = 10
fnx = lambda : np.random.randint(5, 50, length)
x = np.arange(length)
y1, y2, y3 = fnx(), fnx(), fnx()
areaLabels=['area1','area2','area3']
fig, ax = plt.subplots()
ax.stackplot(x, y1, y2, y3)

loc = y1.argmax()
ax.text(loc, y1[loc]*0.25, areaLabels[0])

loc = y2.argmax()
ax.text(loc, y1[loc] + y2[loc]*0.33, areaLabels[1])

loc = y3.argmax()
ax.text(loc, y1[loc] + y2[loc] + y3[loc]*0.75, areaLabels[2]) 

plt.show()

which in test runs is okayish:

enter image description here

enter image description here

enter image description here

Finding the best loc could be fancier -- maybe one wants the x_n, x_(n+1) with the highest average value.

Post a Comment for "Label Areas In Python Matplotlib Stackplot"