https://remp3indir.net/python-plot-color-code/