使用补丁自定义散点图中的图例标记面颜色

Posted 2023-02-16

【中文标题】使用补丁自定义散点图中的图例标记面颜色

【英文标题】：Customize legend marker facecolor in scatterplot with patches

【发布时间】：2020-10-27 21:07:15

【问题描述】：

我有一个散点图，其中还包含一个椭圆。但是，图例仅包含椭圆的格式。我怎样才能反映散点图中的格式？

在下面的示例中，我希望像散点图中一样填充图例标记。

x = np.random.randn(60) 
y = np.random.randn(60)

fig,ax=plt.subplots()
sb.scatterplot(x, y,color='red', ax=ax)
ellipse = matplotlib.patches.Ellipse( 
                xy=(0, 0), width=2, height=3, angle=45, 
                edgecolor='red', facecolor='none'
            )
ax.add_patch(ellipse)

ax.legend(['A'])
plt.show()

【参考方案1】：

Matplotlib 的标准方式是为每个项目分配一个标签以显示在图例中。只有在标准方式不能给出预期结果的情况下，图例才可以是customized。

不带参数调用ax.legend() 会创建默认图例。可以使用Patch' and Line2D` 创建自定义句柄，如下面的代码所示。

import matplotlib.pyplot as plt
from matplotlib.patches import Ellipse
from matplotlib.patches import Patch
from matplotlib.lines import Line2D
import numpy as np
import seaborn as sns

x = np.random.randn(60)
y = np.random.randn(60)

fig, ax = plt.subplots()
sns.scatterplot(x, y, color='red', ax=ax, label='scatter dots')
ellipse = Ellipse(xy=(0, 0), width=2, height=3, angle=45,
                  edgecolor='red', facecolor='none', label='ellipse')
ax.add_patch(ellipse)

handles, labels = ax.get_legend_handles_labels()
new_handles = [Patch(facecolor='white', edgecolor='red', hatch='ooo'),
               Line2D([0], [0], marker='o', markerfacecolor='red', markeredgecolor='black', markersize=10, ls='')]
ax.legend(new_handles, labels)
plt.show()

【讨论】：

由于某种原因，它不适用于我的真实数据，但工作示例完美运行。话虽如此，有没有办法在绘图函数之外独立更改图例中标记的格式（facecolor，edgecolor）