import matplotlib.pyplot as plt
cm = plt.cm.get_cmap('RdYlBu')
xy = list(range(20))
z = xy
sc = plt.scatter(xy, xy, c=z, vmin=0, vmax=20, s=35, cmap=cm)
plt.colorbar(sc)
plt.show()
цветная полоса matplotlib для разброса