python画散点图图例

可以使用Matplotlib库的scatter函数来画散点图，并且可以使用legend函数来添加图例。如下代码：

import matplotlib.pyplot as plt
import random

# 生成随机数据
x = [random.randint(0, 50) for _ in range(20)]
y = [random.randint(0, 50) for _ in range(20)]
colors = ['red' if i < 10 else 'blue' for i in range(20)]

# 画散点图
plt.scatter(x, y, c=colors, marker='o')

# 添加图例
plt.legend(['Category1', 'Category2'])

plt.show()

这段代码会生成一个随机颜色的散点图，并在图例中添加两个类别，命名为Category1和Category2。