使用存储在数据框中的颜色制作散点图和色点

时间:2016-12-09 22:12:56

标签: python pandas visualization

我有一个pandas数据框,其中包含一个存储颜色的列。每行都有具有相关颜色的数据。我想制作数据的散点图,以便根据存储在同一行中的颜色对每个点进行着色。以下是我的数据框的示例:

 import pandas as pd
 df = pd.DataFrame({'dataX': [3,79,90], 'dataY': [7,9,13], 'color': ['Blue', 'Green', 'Red']})

    color  dataX  dataY
 0   Blue      3      7
 1  Green     79      9
 2    Red     90     13

因此索引0处的点将为蓝色,绿色处于索引1,依此类推。

提前致谢!

1 个答案:

答案 0 :(得分:4)

传递颜色参数c

df.plot.scatter('dataX', 'dataY', c=df['color'])

enter image description here