Plotly express colorscale. For the colorbar, you can only use fig.