Matplotlib:从头开始制作彩色标记图例

Posted

技术标签:

【中文标题】Matplotlib:从头开始制作彩色标记图例【英文标题】:Matplotlib : making a colored markers legend from scratch 【发布时间】:2018-05-03 15:41:28 【问题描述】:

在 Matplotlib 中,我正在尝试用像这样的彩色“标记”制作一个图例:

这个是使用scatter 函数制作的,但不适合我的情节。我想“从头开始”制作一个图例,没有相关数据。 颜色很重要,因此应该是每个标记的属性。

我试过了

import matplotlib.markers as mmark
list_mak = [mmark.MarkerStyle('.'),mmark.MarkerStyle(','),mmark.MarkerStyle('o')]
list_lab = ['Marker 1','Marker 2','Marker 3']

plt.legend(list_mak,list_lab)

但是:

1) MarkerStyle 类不支持颜色信息

2) 我收到警告:

UserWarning: Legend does not support <matplotlib.markers.MarkerStyle object at 0x7fca640c44d0> instances.
A proxy artist may be used instead.

但是如何根据标记定义代理艺术家?

感谢您的帮助!

【问题讨论】:

你读过图例指南吗? matplotlib.org/users/… 当然可以,但没有明确引用“标记”图例 那里有一个例子,展示了如何使用line2D 对象添加蓝色星。没有理由你不能对每个标记都做同样的事情(只有一个标记,没有线条)。 【参考方案1】:

按照legend guide 中给出的示例,您可以使用Line2D 对象代替marker 对象。

与指南中给出的示例的唯一区别是您要设置linestyle='None'

import matplotlib.lines as mlines
import matplotlib.pyplot as plt

blue_star = mlines.Line2D([], [], color='blue', marker='*', linestyle='None',
                          markersize=10, label='Blue stars')
red_square = mlines.Line2D([], [], color='red', marker='s', linestyle='None',
                          markersize=10, label='Red squares')
purple_triangle = mlines.Line2D([], [], color='purple', marker='^', linestyle='None',
                          markersize=10, label='Purple triangles')

plt.legend(handles=[blue_star, red_square, purple_triangle])

plt.show()

【讨论】:

这是一个很好的技巧,但使用名为“Line2D”的东西有点误导 在 matplotlib 中,“标记”只是在 plt.plotplt.scatter 函数中使用的东西。 plt.plot 创建 Line2D 对象,plt.scatter 创建 PathCollection 对象。如果您想在图例中添加标记,则基本上必须使用这两个对象之一。我发现Line2D 在这种情况下更容易使用。这不是“技巧”,只是完成您所要求的工作的一种方式。 确实是完成这项工作的好方法!【参考方案2】:

您可以将HandlerBase 子类化以从(color, marker) 的元组创建处理程序。

import matplotlib.pyplot as plt
from matplotlib.legend_handler import HandlerBase

list_color  = ["c", "gold", "crimson"]
list_mak    = ["d","s","o"]
list_lab    = ['Marker 1','Marker 2','Marker 3']

ax = plt.gca()

class MarkerHandler(HandlerBase):
    def create_artists(self, legend, tup,xdescent, ydescent,
                        width, height, fontsize,trans):
        return [plt.Line2D([width/2], [height/2.],ls="",
                       marker=tup[1],color=tup[0], transform=trans)]


ax.legend(list(zip(list_color,list_mak)), list_lab, 
          handler_map=tuple:MarkerHandler()) 

plt.show()

【讨论】:

以上是关于Matplotlib:从头开始制作彩色标记图例的主要内容,如果未能解决你的问题,请参考以下文章

Matplotlib 图例:标记上方的标签

Matplotlib:在图例中移动标记位置

matplotlib 图例标记仅一次

Matplotlib 获取图例标记的坐标

通过 matplotlib 图例中的标记删除线

通过 matplotlib 图例中的标记删除线