如何在Python的ggplot中为双geom_point散点图添加图例

看起来你遇到的问题核心是没有把颜色映射到一个分组变量——如果直接给两个geom_point硬编码颜色（比如color='red'），ggplot不会自动生成图例，因为它没法识别这是一组相关的类别。下面给你两种可行的解决思路，根据你的需求选就行：

方法1：推荐——用单个geom_point，映射分组变量到color #

这是最简洁的方式，适合大多数场景。假设你的数据里有一列（比如叫group）标记每个点是yes还是no：

from plotnine import ggplot, aes, geom_point, scale_color_manual, labs

# 示例数据（替换成你的真实数据）
import pandas as pd
df = pd.DataFrame({
    'x': [1,2,3,4,5],
    'y': [2,4,1,5,3],
    'group': ['yes', 'yes', 'no', 'yes', 'no']
})

# 绘图代码
(ggplot(df, aes(x='x', y='y', color='group'))
 + geom_point(size=3)  # 可根据需求调整点的大小
 + scale_color_manual(values={'yes': '#ff4444', 'no': '#33b5e5'})  # 自定义颜色
 + labs(color='Category')  # 设置图例标题，替换成你想要的名称
)

原理：把group列映射到color美学后，ggplot会自动识别这是一个分类变量，生成对应的图例，再用scale_color_manual指定每个类别对应的颜色即可。

方法2：必须用两个geom_point时的处理 #

如果因为特殊需求（比如要给yes/no设置不同的形状、大小）必须分开用两个geom_point，那需要在每个图层的aes里显式指定color的类别名称，再统一设置颜色映射：

(ggplot(df)
 # 第一个图层：yes组，指定color='yes'（在aes里！）
 + geom_point(aes(x='x', y='y', color='yes'), 
              data=df[df['group']=='yes'], 
              size=3, shape='circle')
 # 第二个图层：no组，指定color='no'（在aes里！）
 + geom_point(aes(x='x', y='y', color='no'), 
              data=df[df['group']=='no'], 
              size=2, shape='triangle')
 # 统一设置颜色和图例
 + scale_color_manual(values={'yes': '#ff4444', 'no': '#33b5e5'}, name='Category')
)

关键注意点：一定要把color='yes'/color='no'放在aes()内部，而不是直接作为geom_point的参数。这样ggplot会把这两个图层的颜色识别为同一个美学维度，从而合并生成图例。

为什么你之前的尝试没成功？ #

如果之前用了scale_color_manual但没显示图例，大概率是因为你把颜色写成了geom_point(color='red')（在aes外面）——这种硬编码的颜色不属于美学映射的一部分，ggplot不会为它生成图例。只有当颜色是映射到变量（或像方法2里的固定类别字符串）时，图例才会出现。

内容的提问来源于stack exchange，提问作者Sparky_47