Matplotlib scatter plot color by value