画散点图时如何根据类别变量进行不同的颜色标注

您可以使用不同的颜色表示不同的类别变量。一种常见的方法是使用 `matplotlib` 库来制作散点图，并使用 `c` 参数指定颜色。例如，您可以使用以下代码将 `x` 和 `y` 数据以不同的颜色绘制：

import matplotlib.pyplot as plt

# 数据
x = [1, 2, 3, 4, 5]
y = [3, 5, 2, 4, 1]
categories = [0, 1, 0, 1, 0]  # 类别变量

# 绘制散点图
plt.scatter(x, y, c=categories)

# 显示图形
plt.show()

在上面的代码中，使用 `c=categories` 参数来指定颜色。它会使用值 `0` 和 `1` 来选择两种不同的颜色。您也可以指定不同的颜色映射，例如 `cmap='coolwarm'`，来使用更多颜色来表示类别变量。