I understand that you want the colorbar to be like image 2 and not as that of image 1.
I am attaching a document link for your reference. It contains the information about the usage of colorbar.
You can also use colorcube instead of colormap to get the desired result. I am attaching a document link which contains about the usage of colorcube.
