Scatter plot with scalar data

Question

I want to create a scatter plot with matplotlib where the data points have scalar data attached to them and they are assigned a color depending on how large their attached value is relative to other points in the set. I want something like a heatmap. However, I am looking for a “discrete” heatmap, i.e. Nothing should be indicated if there were no points in the original dataset, and in particular interpolation (in space) should not be performed.

Can this be done?

Answer 1

you can use the scatter and set the applied value to the c parameter:

import numpy as np import pylab as pl x = np.random.uniform(-1, 1, 1000) y = np.random.uniform(-1, 1, 1000) z = np.sqrt(x*x+y*y) pl.scatter(x, y, c=z) pl.colorbar() pl.show() 

Answer 2

The solution to this is Altair.

 import numpy as np import pylab as pl x = np.random.uniform(-1, 1, 1000) y = np.random.uniform(-1, 1, 1000) z = np.sqrt(x*x+y*y) df = pd.DataFrame({'x':x,'y':y, 'z':z}) from altair import * Chart(df).mark_circle().encode(x='x',y='y', color='z')