如何将颜色作为第三维添加到 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)
结果:
我有一个 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)
结果: