python在scatter按颜色添加图例

在Python的Matplotlib库中，如果你想在一个散点图(scatter plot)中按颜色添加图例，可以使用`plt.scatter()`函数创建散点，并通过设置`c`参数指定颜色，然后使用`plt.legend()`添加图例。例如：

import matplotlib.pyplot as plt

# 数据
x = [1, 2, 3, 4, 5]
y = [10, 8, 6, 4, 2]
colors = ['red', 'green', 'blue', 'yellow', 'purple']  # 颜色列表

# 创建散点图，颜色根据colors列表变化
plt.scatter(x, y, c=colors)

# 添加图例
plt.colorbar(label='颜色类别')  # 使用colorbar创建并标签颜色图例
plt.title('按颜色分类的散点图')
plt.xlabel('X轴')
plt.ylabel('Y轴')

# 显示图例
plt.legend(title='Legend Title', handles=plt.cm.get_cmap('viridis')(range(len(colors))), labels=colors)  # 如果你想自定义图例标题和标签

# 显示图形
plt.show()

在这个例子中，`plt.cm.get_cmap('viridis')(range(len(colors)))`返回一个颜色映射，`handles`用于指定颜色条图例的标记，`labels`则是对应的标签。