颜色散点图基于第三列中的值吗？

Question

我目前正在绘制基于两列数据的散点图。但是，我想根据第三列中的类标签给数据点着色。

我的第三列中的标签是1,2或3.我如何根据第三列中的值对散​​点图着色？

plt.scatter(waterUsage['duration'],waterUsage['water_amount']) 
plt.xlabel('Duration (seconds)') 
plt.ylabel('Water (gallons)') 

Answer 1

scatter函数愉快地获取表示颜色的数字列表。您可以用颜色表玩也一样，如果你想要的（但你不必）：

plt.scatter(waterUsage['duration'], waterUsage['water_amount'],\ 
      c=waterUsage['third_column'], cmap=plt.cm.autumn) 

Answer 2

另一条目添加到字典“色”

def addcolor(b): 

    a=b 
    for x in range(len(a['third_column'])): 
     if a['third_column'][x]==1: a['color'][x]='rosybrown' 
     elif a['third_column'][x]==2: a['color'][x]='papayawhip' 
     elif a['third_column'][x]==3: a['color'][x]='chartreuse' 
    return a 

waterUsage = addcolor(waterUsage) 

plt.scatter(waterUsage['duration'], 
      waterUsage['water_amount'], 
      c=waterUsage['color']) 

matplotlib接受灰度，RGB ，十六进制和HTML颜色：

http://matplotlib.org/api/colors_api.html

HTML颜色列表，按组：

https://www.w3schools.com/colors/colors_groups.asp