Python - 使用查找字典设置pyplot散射的颜色

Question

我正在尝试使用来自熊猫数据帧的数据为散点图的点着色，如下所示：

x = [5,7,12,0,19,56,40,16,36,17,81,50]
y = [20,26,37,40,8,63,13,28,47,91,26,2]
letter = ['A','B','C','D','E','F','E','B','A','C','F','D']
df = pd.DataFrame({'x' : x, 'y' : y,'letter' : letter})

我还设置了一个查找字典，以决定将哪种颜色分配给"字母"列中的哪个值：

letter_col = {'A':'blue', 'B':'red', 'C':'green', 'D':'yellow', 'E':'orange', 'F':'pink'}

使用 pyplot 散射函数的颜色参数 （c=），如何使用 letter_col 设置点的颜色？

Answer 1

我只是在数据帧内创建一个color列，然后将其传递给散点函数

x = [5,7,12,0,19,56,40,16,36,17,81,50]
y = [20,26,37,40,8,63,13,28,47,91,26,2]
letter_col = {'A':'blue', 'B':'red', 'C':'green', 'D':'yellow', 'E':'orange', 'F':'pink'}
letter = ['A','B','C','D','E','F','E','B','A','C','F','D']
colors = map(lambda x: letter_col.get(x, 'black'), letter)
df = pd.DataFrame({'x' : x, 'y' : y,'letter' : letter, 'color': colors})