Re: [Matplotlib-users] Individual custom markers and colorbar

[Ryan N. <rne...@gm...>]

Hackstein,

Unfortunately, I'm not sure of an 'elegant' way to do what your asking 
with a single call to scatter. Others may know a better way. However, 
you can use rectangle patches and patch collections. (Requires a bit 
more code than scatter but is ultimately more flexible.)

I think the example below does what you need, but with random numbers.

Hope it helps a little.

Ryan

#######################
import numpy as np
import matplotlib.pyplot as plt
from matplotlib.patches import Rectangle
from matplotlib.collections import PatchCollection

n = 100

# Get your xy data points, which are the centers of the rectangles.
xy = np.random.rand(n,2)

# Set a fixed height
height = 0.02
# The variable widths of the rectangles
widths = np.random.rand(n)*0.1

# Get a color map and color values (normalized between 0 and 1)
cmap = plt.cm.jet
colors = np.random.rand(n)

rects = []
for p, w, c in zip(xy, widths, colors):
     xpos = p[0] - w/2 # The x position will be half the width from the 
center
     ypos = p[1] - height/2 # same for the y position, but with height
     rect = Rectangle( (xpos, ypos), w, height ) # Create a rectangle
     rects.append(rect) # Add the rectangle patch to our list

# Create a collection from the rectangles
col = PatchCollection(rects)
# set the alpha for all rectangles
col.set_alpha(0.3)
# Set the colors using the colormap
col.set_facecolor( cmap(colors) )

# Make a figure and add the collection to the axis.
ax = plt.subplot(111)
ax.add_collection(col)
plt.show()

###############################


On 4/24/2013 5:35 PM, Hackstein wrote:
>
> Hi all,
>
> I am trying to get a scatter plot using a colormap. Additionally, I 
> need to define every marker for every data point individually -- each 
> being a rectangle with fixed height but varying width as a function of 
> the y-value. X and y being the data coordinates, z being a number to 
> be color coded with the colormap.
>
> Ideally, I would like to create a list of width and height values for 
> each data point and tell the scatter plot to use those.
>
> So far I got colormapped data with custom markers (simplified):
>
> [code]
>
> import numpy as np
>
> import matplotlib.pyplot as plt
>
> from pylab import *
>
> x = y = [1,2,3,4,5]
>
> z = [2,4,6,8,10]
>
> colors = cm.gnuplot2
>
> verts_vec = list(zip([-10.,10.,10.,-10.],[-5.,-5.,5.,5.]))
>
> fig = plt.figure(1, figsize=(14.40, 9.00))
>
> ax = fig.add_subplot(1,1,1)
>
> sc = ax.scatter(x, y, c=np.asarray(z), marker=None, edgecolor='None', 
> verts=verts_vec, cmap=colors, alpha=1.)
>
> plt.colorbar(sc, orientation='horizontal')
>
> plt.savefig('test.png', dpi=200)
>
> plt.close(1)
>
> [/code]
>
> But I need to define a marker size for each point, and I also need to 
> do that in axis scale values, not in points.
>
> I imagine giving verts a list of N*2 tuples instead of 2 tuples, N 
> being len(x), to define N individual markers.
>
> But when doing that I get the error that vertices.ndim==2.
>
> A less elegant way would be to plot every data point in an individual 
> scatter plot function, using a for-loop iterating over all data 
> points. Then, however, I see no way to apply a colormap and colorbar.
>
> What is the best way to accomplish that then?
>
> Thanks,
>
> -Hackstein
>
>
>
> ------------------------------------------------------------------------------
> Try New Relic Now & We'll Send You this Cool Shirt
> New Relic is the only SaaS-based application performance monitoring service
> that delivers powerful full stack analytics. Optimize and monitor your
> browser, app, & servers with just a few lines of code. Try New Relic
> and get this awesome Nerd Life shirt! http://p.sf.net/sfu/newrelic_d2d_apr
>
>
> _______________________________________________
> Matplotlib-users mailing list
> Mat...@li...
> https://lists.sourceforge.net/lists/listinfo/matplotlib-users