nice improvements. I got one big problem with the colorbar:
How do I set a ylabel to the colorbar?
Another idea for NaNs would be to just leave out the text "NaN" as well, simply don't show any text in those fields.
Something else I thought of:
the y-labels of the colorbar are always set automatically (or is there a way I don't see?). One tick is close to the top, as it should be, yet there is no label close to the bottom of the colorbar. This is not nice. They should be evenly spread across the colorbar with one close to the top value and one close to the bottom value. :)
Yes this is possible, but there are two caveats:
1. You first need a criterion of your own for determining what defines an outlier.
2. You have to do a *little* extra coding to highlight the points. Here is an example where we highlight the most positive point only. This is the case because I deliberately have avoided adding too many extra features such as this into the function. The idea is to make it easy for others to modify the plots as needed.
Here is a toy example: