utils: transform the default plot explanation

[lucashervier]

• Change the default plot to get "prettier" results.
• Separate plot function and the preparation of the plot (i.e return the ready to plot array)
• Handle Nan, Inf values