matplotlib 合并两个Pyplot图块以生成图例

368yc8dk  于 2023-01-21  发布在  其他
关注(0)|答案(4)|浏览(159)

我试图用置信带绘制一些数据。我为每个数据流绘制了两个图:plotfill_between。我希望图例看起来与图类似,其中每个条目都有一个框(置信区域的颜色),带有一条穿过中心的较暗实线。到目前为止,我已经能够使用面片创建矩形图例键,但我不知道如何实现中心线。我尝试使用影线,但无法控制位置。厚度或颜色。
我最初的想法是尝试并合并两个补丁(补丁和二维线);然而,它还没有奏效。有更好的方法吗?我的MWE和目前的数字如下所示。

import matplotlib.patches as mpatches
import matplotlib.pyplot as plt
import numpy as np

x = np.linspace(0,1,11)
y = np.linspace(0,1,11)

plt.plot(x, y, c='r')
plt.fill_between(x, y-0.2, y+0.2, color='r', alpha=0.5)
p = mpatches.Patch(color='r', alpha=0.5, linewidth=0)

plt.legend((p,), ('Entry',))

7rfyedvj

7rfyedvj1#

该解决方案是从CrazyArm的评论中借用的,可以在这里找到:显然,你可以列出一个句柄列表,只分配一个标签,它神奇地将两个句柄/艺术家结合在一起。

import matplotlib.patches as mpatches
import matplotlib.pyplot as plt
import numpy as np

x = np.linspace(0,1,11)
y = np.linspace(0,1,11)

p1, = plt.plot(x, y, c='r')  # notice the comma!
plt.fill_between(x, y-0.2, y+0.2, color='r', alpha=0.5)
p2 = mpatches.Patch(color='r', alpha=0.5, linewidth=0)

plt.legend(((p1,p2),), ('Entry',))

mrzz3bfm

mrzz3bfm2#

从代码开始
这是最接近我可以得到你。可能有一种方法来创建补丁的方式,你想要的,但我是一个有点新的,以及所有我能做的是建立适当的图例:

import matplotlib.patches as mpatches
import matplotlib.pyplot as plt
import numpy as np

x = np.linspace(0,1,11)
y = np.linspace(0,1,11)

fig = plt.figure()
ax = fig.add_subplot(111)
plt.plot(x, y, c='r',label='Entry')
plt.fill_between(x, y-0.2, y+0.2, color='r', alpha=0.5)
p_handle = [mpatches.Patch(color='r', alpha=0.5, linewidth=0)]
p_label = [u'Entry Confidence Interval']
handle, label = ax.get_legend_handles_labels()
handles=handle+p_handle
labels=label+p_label
plt.legend(handles,labels,bbox_to_anchor=(0. ,1.02 ,1.,0.3),loc=8,
           ncol=5,mode='expand',borderaxespad=0,prop={'size':9},numpoints=1)

plt.show()

从我可以告诉你将不得不创建一个“艺术家”的对象,适合你正在寻找的设计,我找不到这样做的方法。一些类似的例子可以在这个线程中找到:Custom Legend Thread
希望这有助于好运,我有兴趣,如果有更深入的方式以及。

sgtfey8w

sgtfey8w3#

我也有“类似”的问题。多亏了这个问题,我才能做到以下几点。

fig = pylab.figure()
figlegend = pylab.figure(figsize=(3,2))
ax = fig.add_subplot(111)
point1 = ax.scatter(range(3), range(1,4), 250, marker=ur'$\u2640$', label = 'S', edgecolor = 'green')
point2 = ax.scatter(range(3), range(2,5), 250, marker=ur'$\u2640$', label = 'I', edgecolor = 'red')
point3 = ax.scatter(range(1,4), range(3),  250, marker=ur'$\u2642$', label = 'S', edgecolor = 'green')
point4 = ax.scatter(range(2,5), range(3), 250, marker=ur'$\u2642$', label = 'I', edgecolor = 'red')
figlegend.legend(((point1, point3), (point2, point4)), ('S','I'), 'center',  scatterpoints = 1, handlelength = 1)
figlegend.show()
pylab.show()

然而,我的两个标记(金星和火星)在图例中重叠。我试着玩手柄长度,但似乎没有帮助。任何建议或意见将是有益的。

3hvapo4f

3hvapo4f4#

对于那些想进一步推动这个主题的人

我遇到过一个非常相似的问题,不同之处在于我希望图例用虚线和填充的矩形表示自定义形状(也是用plotfill_between创建的)(因为该形状具有此模式)。

我找到的解决方案是按照Matplotlib Legend指南实现一个自定义图例处理程序。
特别是,我实现了一个定义legend_artist方法的类,该方法允许您通过向handlebox添加补丁来尽可能多地定制处理程序(或者这是我的理解)。
我用矩形而不是斑点形状发布了一个MWE(这需要很大的空间,而且会模糊信息)。

import matplotlib.patches as mpatches
import matplotlib.pyplot as plt

class DashedFilledRectangleHandler:
    def legend_artist(self, legend, orig_handle, fontsize, handlebox):
        x0, y0 = handlebox.xdescent, handlebox.ydescent
        width, height = handlebox.width, handlebox.height
        patch_fill = mpatches.Rectangle((x0, y0), width, height, facecolor='tab:blue', linestyle='', alpha=0.1,
                                        transform=handlebox.get_transform())
        patch_line = mpatches.Rectangle((x0, y0), width, height, fill=False, edgecolor='tab:blue', linestyle='--',
                                        transform=handlebox.get_transform())
        handlebox.add_artist(patch_line)
        handlebox.add_artist(patch_fill)
        return patch_line

fig, ax = plt.subplots()

rectangle_fill = ax.add_patch(mpatches.Rectangle((0, 0), 2, 1, facecolor='tab:blue', edgecolor='tab:blue', linestyle='',
                                                 alpha=0.1))
rectangle_line = ax.add_patch(mpatches.Rectangle((0, 0), 2, 1, fill=False, edgecolor='tab:blue', linestyle='--',
                                                 lw=3))

ax.legend([rectangle_fill], ['Dashed filled rectangle handler'],
          handler_map={rectangle_fill: DashedFilledRectangleHandler()})
ax.set_xlim([-1, 3])
ax.set_ylim([-1, 2])
plt.show()

相关问题