9

我正在使用 matplotlib 进行一些绘图,并且我有一个图例,它告诉查看器这些点是用哪些传感器记录的。有多种类型的传感器,我想在图例中加上字幕,告诉观众每组是哪种传感器。我有一个可行的解决方案,但它有点像 hack,如下所示:

在此处输入图像描述

创建图例时,它接受两个重要参数:图例标记列表和图例标签列表。我目前的解决方案是将字幕标记设置为带有白色轮廓的白色框,并让字幕标签被两个换行符包围。看起来还可以,但是如果字幕不缩进的话会显得专业很多。我尝试过的两种解决方法是将字幕的标记设置为无,并将字幕的标记设置为所需的字幕字符串,并将其标签设置为空字符串。两者都没有工作。有人对这个有经验么?非常感谢。

4

1 回答 1

9

我能想到的最好的办法是为字符串制作一个自定义处理程序。

import matplotlib.pyplot as plt
import matplotlib.text as mtext


class LegendTitle(object):
    def __init__(self, text_props=None):
        self.text_props = text_props or {}
        super(LegendTitle, self).__init__()

    def legend_artist(self, legend, orig_handle, fontsize, handlebox):
        x0, y0 = handlebox.xdescent, handlebox.ydescent
        title = mtext.Text(x0, y0, r'\underline{' + orig_handle + '}', usetex=True, **self.text_props)
        handlebox.add_artist(title)
        return title


[line1] = plt.plot(range(10))
[line2] = plt.plot(range(10, 0, -1), 'o', color='red')
plt.legend(['Title 1', line1, 'Title 2', line2], ['', 'Line 1', '', 'Line 2'],
           handler_map={basestring: LegendTitle({'fontsize': 18})})

plt.show()

输出

我基于http://matplotlib.org/users/legend_guide.html中的示例。

于 2016-07-20T16:37:29.920 回答