7

我想在 matplotlib 堆栈图的区域内生成标签。我会满足于标记用于界定该区域的线。考虑这个例子:

import numpy as np
from matplotlib import pyplot as plt

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

这会产生:在此处输入图像描述

但我想制作这样的东西:

在此处输入图像描述

matplotlib 等高线图具有这种类型的标记功能(尽管在等高线图的情况下对线进行了标记)。

感谢您提供任何帮助(甚至重定向到我可能错过的帖子)。

4

1 回答 1

5

啊,启发式。像这样的东西?:

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()

在测试运行中是好的:

在此处输入图像描述

在此处输入图像描述

在此处输入图像描述

找到最好的loc可能会更有趣——也许有人想要x_n, x_(n+1)平均值最高的。

于 2015-07-22T21:36:55.257 回答