Il te suffit de changer la colormap (par exemple tape pyplot.hot()).