Re: [Matplotlib-users] Single Point Stored Plot

>>>>> "andrea" == andrea gavana@tin it <and...@ti...> writes:

    andrea> Hello NG, I am facing a small problem for which I was
    andrea> unable to find a solution. Basically, I have to make a
    andrea> scatter plot of x, y coordinates (they represents oil
    andrea> wells positions) on a 2D map, but I have some constraints:

    andrea> 1) I would like to have each point as a separate "plot
    andrea> object", because the user may decide to hide/show only a
    andrea> particular well, or show/hide them all, or show/hide
    andrea> whatever combination of wells.  2) At the moment when the
    andrea> class that holds a single well position/status is created,
    andrea> the plot window may not be there (because is the user that
    andrea> decides when to open a new plot window).

    andrea> I have tried something like:

Hey Andrea, sorry for the delay.  This is a neat problem and one that
collections were designed to solve.  You probably want to use a
RegularPolyCollection rather than a PolyCollection (the latter is for
arbitrary polygons and you must compute all the vertices yourself).
Here is an example that allows you to add and delete markers -- it
inserts them with random positions and colors.  You should be able to
adapt it to your application quite easily.

import random
from matplotlib.colors import colorConverter
from matplotlib.collections import RegularPolyCollection 
import matplotlib.cm as cm
from pylab import figure, show, nx

fig = figure()
ax = fig.add_subplot(111, xlim=(0,1), ylim=(0,1), autoscale_on=False)
ax.set_title("Press 'a' to add a point, 'd' to delete one")
# a single point
offsets = [(0.5,0.5)]
facecolors = [cm.jet(0.5)]

collection = RegularPolyCollection(
    fig.dpi,
    numsides=5, # a pentagon
    rotation=0,
    sizes=(50,), 
    facecolors = facecolors,
    edgecolors = (colorConverter.to_rgba('black'),),
    linewidths = (1,),
    offsets = offsets,
    transOffset = ax.transData,
    )

ax.add_collection(collection)

def onpress(event):
    """
    press 'a' to add a random point from the collection, 'd' to delete one
    """
    if event.key=='a':
        x,y = nx.mlab.rand(2)
        color = cm.jet(nx.mlab.rand())
        offsets.append((x,y))
        facecolors.append(color)
        fig.canvas.draw()
    elif event.key=='d':
        N = len(offsets)
        if N>0:
            ind = random.randint(0,N-1)
            offsets.pop(ind)
            facecolors.pop(ind)
            fig.canvas.draw()

fig.canvas.mpl_connect('key_press_event', onpress)

show()