Question

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

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})

我还设置了一个查找字典，以决定将哪种颜色分配给＆＃39;字母中的哪个值。柱：

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})

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

1 个答案: