Python:如何在绘制散点图时区分类别值？

import matplotlib.pyplot as plt
import numpy as np

# 生成随机数据
np.random.seed(0)
x = np.random.randn(100)
y = np.random.randn(100)
categories = np.random.randint(0, 3, 100)  # 生成0、1、2三个分类别值

# 绘制散点图
plt.scatter(x[categories==0], y[categories==0], c='r', marker='o', label='Category 0')
plt.scatter(x[categories==1], y[categories==1], c='g', marker='s', label='Category 1')
plt.scatter(x[categories==2], y[categories==2], c='b', marker='^', label='Category 2')

# 添加图例和标签
plt.legend()
plt.xlabel('X')
plt.ylabel('Y')

# 显示图形
plt.show()