Matplotlib 表格图例

Question

我有许多数据集正在尝试绘制，我正在寻找一种表示图例的表格方式。我正在生成的情节是以下形式：

mul_list = [1,2,3]
add_list = [[1,2,3],[4,5,6]]

linestyles = [':','--','-']
linecolors = ['r','g']

time = np.linspace(0,5)

for i,mul in enumerate(mul_list):
    for j,add in enumerate(add_list):
        plt.plot(time,mul*time+add[i],linestyle = linestyles[i],color = linecolors[j])

我正在尝试生成表单的图例：

------------------------------
|               'r'     'g'  |
| ':'  m = 1   a = 1   a = 4 |
|'--'  m = 2   a = 2   a = 5 |
| '-'  m = 3   a = 3   a = 6 |
------------------------------

其中 'r','g' 将是具有红色和绿色颜色的标记，而 ':','--','-' 将是具有正确样式的黑色标记。这是可行的吗？我环顾四周，并没有找到一种以这种方式在图例上放置两组独立标记的好方法。

Answer 1

如果允许有细微的变化，您可以为每个组合创建标签并使用多列：

from matplotlib import pyplot as plt

mul_list = [1, 2, 3]
add_list = [[1, 2, 3], [4, 5, 6]]

linestyles = [':', '--', '-']
linecolors = ['r', 'g']

time = np.linspace(0, 5)

for i, mul in enumerate(mul_list):
    for j, add in enumerate(add_list):
        plt.plot(time, mul * time + add[i], linestyle=linestyles[i], color=linecolors[j], label=f'm={mul} a={add[i]}')
plt.legend(ncol=len(mul_list))
plt.show()

以不同的顺序绘制（或手动重新排列句柄和标签）可以给出两列：

for j, add in enumerate(add_list):
    for i, mul in enumerate(mul_list):
        plt.plot(time, mul * time + add[i], linestyle=linestyles[i], color=linecolors[j], label=f'm={mul} a={add[i]}')
plt.legend(ncol=len(add_list))

如果您真的想要更复杂的图例，另请参阅 Table legend in matplotlib。