如何根据类别列表为 MatPlotLib 散点图着色

Question

我有一个名为 DATA 的 Pandas 数据框，我将其传递给一个函数以对数据中的两列进行分类（请参见下面的代码）。结果是一个字符串列表，其长度与 DATA 相同。该列表与数据框是分开的（我必须这样做，因为它用于分配）。

我正在尝试创建一个 MatPlotLib 散点图，将 DATA 中的两列绘制为 x 和 y 值：

x = DATA.loc[:]['A']
y = DATA.loc[:]['B']

我希望情节点根据相应的类别进行着色（即“健康”为绿色，“不健康”为红色）我还必须包括这些属性：

 alpha = 0.5, s = 100

我将如何去做呢？

我尝试使用颜色映射将类别的名称映射到颜色，然后在类别列表本身上调用 .map，这不起作用，因为列表没有该方法。有点只是从网站上拿来的。

fig, ax = plt.subplots()
colors = {"Healthy": "green", "Unhealthy": "blue", "Morbid": "red"}
ax.scatter(x, y, c=[CATEGORIES].map(colors))