如何将颜色作为第三维添加到 matplotlib 散点图?

How do I add color as a 3rd dimension to matplotlib scatterplot?

我有一个 pandas 数据框,如下所示:

类别X_AxisY_Axis
类别 1 10000 10000
类别 2 15000 20000
类别 3 20000 30000
类别 1 25000 40000
类别 2 10000 50000

我希望我的代码创建一个散点图,水平方向为 X_Axis,垂直方向为 Y_Axis,颜色显示类别,每个类别都有不同的颜色和图例显示他们。

由于数据的大小和很多不同的类别,我无法手动执行此操作。

提前致谢。

如果可以使用seaborn:

import seaborn as sns

sns.scatterplot('X_Axis','Y_Axis',hue='Category',data=df)

结果: