https://stats-learn.com/en/python/the-way-to-colour-a-scatterplot-via-worth/