I have a 2D point cloud that I displayed with scatter and a probability distribution on these points. Is there a quick way on Python to colour the points in relation to my d