matplotlib.pyplot
matplotlib.pyplot.viridis
matplotlib.pyplot.
viridis
Set the colormap to "viridis".
This changes the default colormap as well as the colormap of the current image if there is one. See help(colormaps) for more information.
help(colormaps)