在第一步基础上用plt.scatter将不同数据按不同颜色分类显示

在Python的数据可视化库matplotlib中，如果你已经完成了数据的初步处理并想根据某些变量将不同的数据点区分开来，你可以使用`scatter`函数，并结合`c`参数进行颜色编码。假设你有一个名为`data`的DataFrame，其中包含两列`x_values`和`y_values`作为坐标，以及一列`category`表示类别，可以这样做：

import matplotlib.pyplot as plt

# 假设data是你存储数据的DataFrame
colors = {'category1': 'red', 'category2': 'blue', 'category3': 'green'}  # 定义不同的颜色映射

plt.figure(figsize=(10, 6))  # 创建一个新的图形窗口

for category, color in colors.items():
    plt.scatter(data[data['category'] == category]['x_values'], data[data['category'] == category]['y_values'],
                c=color, label=category)  # 根据类别选择颜色并绘制散点图

plt.title('不同类别的数据点')
plt.xlabel('x_values')
plt.ylabel('y_values')
plt.legend()  # 显示图例，标识每个颜色代表的类别
plt.show()