Color scatter plot points based on a value in third column?

python matplotlib scatter-plot
Gary picture Gary · Mar 8, 2017 · Viewed 16k times · Source

I am currently plotting a scatterplot based on two columns of data. However, I would like to color the datapoints based on a class label that I have in a third column.

The labels in my third column are either 1,2 or 3. How would I color the scatter plot points based on the values in this third column?

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

Answer

DYZ picture DYZ · Mar 8, 2017

The scatter function happily takes a list of numbers representing color. You can play with a colormap, too, if you want (but you don't have to):

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

Related questions

Setting different color for each series in scatter plot on matplotlib

Suppose I have three data sets: X = [1,2,3,4] Y1 = [4,8,12,16] Y2 = [1,4,9,16] I can scatter plot this: from matplotlib import pyplot as plt plt.scatter(X,Y1,color='red') plt.scatter(X,Y2,color='blue') plt.show() How can I do this with 10 …

Read More
Matplotlib scatter plot with different text at each data point

I am trying to make a scatter plot and annotate data points with different numbers from a list. So, for example, I want to plot y vs x and annotate with corresponding numbers from n. y = [2.56422, 3.77284, 3.52623, 3.51468, 3.02199] z = [0.15, 0.3, 0.45, 0.6, 0.75] n = [58, 651, 393, 203, 123] ax = fig.…

Read More
How to do a scatter plot with empty circles in Python?

In Python, with Matplotlib, how can a scatter plot with empty circles be plotted? The goal is to draw empty circles around some of the colored disks already plotted by scatter(), so as to highlight them, ideally without having to …

Read More