I am trying to create a plot with a Colorbar.I am using the shared code from Shan Dou in this post: Matplotlib - add colorbar to a sequence of line plots
The code is