Python颜色在散点图中分类

我想散点图2个变量，并根据第三个变量给点上色，这是一个包含2类的定性变量

df.plot（kind='scatter'，x='TotalIncome'，y='LoanAmount'，figsize=（10,6））
plt.xlabel（“总收入”）
plt.ylabel（“贷款金额”）
plt.show（）

直接根据第三个变量为各个点上色。您只需要将这个定性变量映射到可以表示颜色的东西。例如，如果第三个变量在以下代码中为“c”：

import matplotlib.pyplot as plt

def map_col(col):

    if col == 'y_odd':
        mapped_col = 0
    elif col == 'y_even':
        mapped_col = 1

    return mapped_col


x = [1, 4, 2, 7, 4, 9]
y = [4, 1, 3, 6, 8, 2]
c = ['y_even', 'y_odd', 'y_odd', 'y_even', 'y_even', 'y_even']

color = [map_col(col) for col in c]

plt.scatter(x, y, s=100, c=color, cmap="gnuplot")

plt.show()

有关可用颜色贴图的列表，请参见：

---

以及如何自定义颜色集您可以使用颜色映射来自定义这些值在绘图上对应的内容您介意使用此颜色映射编辑原始代码吗please@MohamedAmine更新了我的答案，添加了彩色地图

sns.scatterplot（data=df，x='TotalIncome'，y='LoanAmount'，hue='OtherColumn'）

？