python 中用户定义的图例

Question

我有这个图，其中曲线之间的一些区域被定义填充。有什么方法可以将它们包含在图例中吗？特别是那些填充区域重叠的地方，以及出现新的不同颜色的地方。

或者有可能定义一个任意的图例而不考虑曲线的数据？

Answer 1

使用 fill_bettween 绘制数据将自动在图例中包含填充区域。

要包括两个数据集重叠的区域，您可以将两个数据集的图例句柄合并为一个图例句柄。

正如评论中指出的，您还可以使用代理定义任意图例句柄。

最后，您可以准确定义要在图例中显示的句柄和标签，而不考虑图表中绘制的数据。

请参阅下面说明上述要点的 MWE：

import matplotlib.pyplot as plt
import numpy as np

plt.close('all')

# Gererate some datas:
x = np.random.rand(50)
y = np.arange(len(x))

# Plot data:
fig, ax = plt.subplots(figsize=(11, 4))
fillA = ax.fill_between(y, x-0.25, 0.5, color='darkolivegreen', alpha=0.65, lw=0)
fillB = ax.fill_between(y, x, 0.5, color='indianred', alpha=0.75, lw=0)

linec, = ax.plot(y, np.zeros(len(y))+0.5, color='blue', lw=1.5)
linea, = ax.plot(y, x, color='orange', lw=1.5)
lineb, = ax.plot(y, x-0.25, color='black', lw=1.5)

# Define an arbitrary legend handle with a proxy:
rec1 = plt.Rectangle((0, 0), 1, 1, fc='blue', lw=0, alpha=0.25)

# Generate the legend:
handles = [linea, lineb, linec, fillA, fillB, (fillA, fillB),
           rec1, (fillA, fillB, rec1)]
labels = ['a', 'b', 'c', 'A', 'B', 'A+B', 'C', 'A+B+C']
ax.legend(handles, labels, loc=2, ncol=4)

ax.axis(ymin=-1, ymax=2)

plt.show()

Answer 2

是的，你完全正确 ian_itor，tacaswell 和 Jean-Sébastien，用户定义的图例似乎是唯一的解决方案，此外我还为 linewidth那些区域要与曲线区分开来，并使用 alpha 得到正确的颜色。

handles, labels = ax.get_legend_handles_labels()
display = (0,1,2,3,4)

overlap_1 = plt.Line2D((0,1),(0,0), color='firebrick', linestyle='-',linewidth=15, alpha = 0.85)
overlap_2= plt.Line2D((0,1),(0,0), color='darkolivegreen',linestyle='-',linewidth=15, alpha = 0.65)
over_lo_3= plt.Line2D((0,1),(0,0), color='indianred',linestyle='-',linewidth=15, alpha = 0.75) 

ax.legend([handle for i,handle in enumerate(handles) if i in display]+[overlap_1 , overlap_2 , overlap_3 ],
      [label for i,label in enumerate(labels) if i in display]+['D','F','G'])

python 中用户定义的图例

User defined legend in python

python

matplotlib

legend-properties