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

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)