Python散点图图例颜色不匹配且生成双图问题求助

嘿，我来帮你搞定这个散点图的问题！你说运行代码后生成两个图，还出现图例颜色不匹配的情况，我来给你分析原因并提供修正方案～

• 生成两个图：大概率是代码里重复调用了plt.show()，或者多次创建了新的绘图轴对象（比如多次用plt.subplots()），导致Matplotlib生成了多个独立的绘图窗口
• 颜色不匹配：通常是手动添加图例时，没有和散点图实际使用的颜色规则对应上，要么是颜色映射逻辑不一致，要么是图例的颜色是手动指定的，和散点的实际颜色脱节

我把你的模拟数据集补全，并调整了绘图逻辑，确保图例颜色和散点完全匹配，且只生成一个图：

import matplotlib.pyplot as plt
import pandas as pd

# 模拟数据集
d = {'vote': [100, 50, 1, 23, 55, 67, 89, 44], 
     'ballot': ['a','Yes','a','No','b','a','a','b'], 
     'whichballot': [1, 2, 1, 1, 2, 1, 1, 2]}
dfwl = pd.DataFrame(d)

# 统一定义颜色映射规则，把whichballot的取值和颜色绑定
color_mapping = {1: '#1f77b4', 2: '#ff7f0e'}

# 创建单个绘图画布和轴，所有绘图都在这个轴上进行
fig, ax = plt.subplots(figsize=(8, 4))

# 按whichballot分组绘制散点，同时为每个分组设置标签
for ballot_type, group_data in dfwl.groupby('whichballot'):
    # 绘制散点，直接用映射好的颜色，同时设置label参数
    ax.scatter(
        x=group_data['vote'], 
        y=range(len(group_data)),  # 这里用索引作为y值，你可以根据需求替换
        color=color_mapping[ballot_type],
        label=f'whichballot = {ballot_type}'
    )

# 直接调用轴对象的legend方法，自动匹配散点的颜色和标签
ax.legend(title='Ballot Type')

# 补充坐标轴标签，让图表更清晰
ax.set_xlabel('Vote Count')
ax.set_ylabel('Data Index')

# 只调用一次plt.show()，确保只弹出一个绘图窗口
plt.show()

• 单轴绘图：用fig, ax = plt.subplots()创建唯一的绘图轴，所有散点都绘制在这个轴上，避免生成多个图
• 统一颜色映射：提前定义color_mapping，让每个whichballot取值对应固定颜色，从根源保证图例和散点颜色一致
• 自动匹配图例：绘图时给每个散点组设置label参数，再用ax.legend()自动生成图例，不需要手动指定颜色，完全避免匹配错误
• 单次显示：只调用一次plt.show()，不会弹出多个窗口

如果你的原代码出现问题，大概率是这几个情况：

• 重复调用了plt.show()，导致弹出多个窗口
• 没有指定绘图轴，多次调用plt.scatter()时Matplotlib自动创建了新的图
• 手动创建图例时（比如plt.legend(['xxx', 'yyy'], colors=['red', 'blue'])），手动指定的颜色和散点实际使用的颜色不对应

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