[Matplotlib-users] plot with marker color coded according to z-value

2012-10-19 Thread elmar werling
Hi, is there a way to adjust the marker color in a xy-plot in relation to the value of a third parameter. Something as the following - not working - example 1. Example 2 is working but rather slow for large arrays. cheers Elmar # example 1 import matplotlib.pyplot as plt x = [1,2,3,4] y

Re: [Matplotlib-users] plot with marker color coded according to z-value

2012-10-19 Thread Joe Kington
That's what ``scatter`` is intended for. Basically, you want something like: plt.scatter(x, y, c=z, marker='s') plt.colorbar() Note that you can also vary the markers by size based on an additional parameter, as well. Have a look at this example:

Re: [Matplotlib-users] plot with marker color coded according to z-value

2012-10-19 Thread elmar werling
thanks for help, finally I found the following solution elmar import numpy as np import matplotlib as mpl import matplotlib.pyplot as plt N = 200 x = np.linspace(0,1,N) y = np.random.randn(N) z = np.random.randn(N)*2+5 cm = mpl.cm.get_cmap('RdYlBu') sc = plt.scatter(x, y, c=z, vmin=min(z),

Re: [Matplotlib-users] plot with marker color coded according to z-value

2012-10-19 Thread Daπid
On Fri, Oct 19, 2012 at 11:08 PM, elmar werling el...@net4werling.de wrote: vmin=min(z), vmax=max(z) A suggestion, when dealing with arrays, it is generally faster to use the numpy function to compute the max and min, either np.max(z) or z.max(), than the standard Python one.

Re: [Matplotlib-users] plot with marker color coded according to z-value

2012-10-19 Thread Damon McDougall
On Fri, Oct 19, 2012 at 10:23 PM, Daπid davidmen...@gmail.com wrote: On Fri, Oct 19, 2012 at 11:08 PM, elmar werling el...@net4werling.de wrote: vmin=min(z), vmax=max(z) A suggestion, when dealing with arrays, it is generally faster to use the numpy function to compute the max and min, either

Re: [Matplotlib-users] plot with marker color coded according to z-value

2012-10-19 Thread elmar werling
Am 19.10.2012 23:26, schrieb Damon McDougall: Correct me if I'm wrong, but I don't even think you need them. I think the default cmap behaviour is to normalise to the min and max of the data. yes, default cmap behaviour will normalise to the min and max of the data.