On Feb 6, 2009, at 8:45 AM, Zunbeltz Izaola wrote:

Dear all,

I would like to have a plot where the frame only have left and
bottom border. I can not find in the documentation any function to draw
the Rectangle contained in figure() only with this 2 lines. It is
possilbe?

Hi Zunbeltz,

Attached is an example of a custom Axes class that does what you want. There are examples at the bottom of the file that show its use. I worked on generalizing this idea for inclusion in MPL, but the code got really nasty, really quickly.

On a side note, if any of the MPL devs think this would make a useful API example (since this topic has come up a few times on the list), feel free to do whatever you want with it.

Cheers,
-Tony

#!/usr/bin/env python
"""
Frame classes for customizing frame borders that surround the plot axes.
"""
import numpy as np

import matplotlib.axes as maxes
import matplotlib.pyplot as plt
import matplotlib.artist as martist
import matplotlib.collections as col
import matplotlib.projections as projections


class Frame(martist.Artist):
    """Draw frame along the edges of the axes patch.
    
    Frame position can be controlled upon initialization or by setting
    `positions` property with a list of positions
        ['left', 'right', 'top', 'bottom' | 'all']
    """
    _position_list = ('left', 'right', 'top', 'bottom')
    def __init__(self, axes, positions=('left', 'bottom'), **kwargs):
        """
        `positions` is a list of strings of frame positions to plot.
            ['left', 'right', 'top', 'bottom' | 'all']
        """
        super(Frame, self).__init__()
        # TODO: allow more keyword configuration
        self.axes = axes
        
        rc = plt.rcParams
        self.color = kwargs.pop('color', rc['axes.edgecolor'])
        self.linewidth = kwargs.pop('linewidth', rc['axes.linewidth'])
        self.linestyle = kwargs.pop('linestyle', 'solid')
        self.positions = positions

    def get_data(self):
        """Convenience method returns tuple of (x, y) data in `self.axes`"""
        x, y = [], []
        ax = self.axes
        for artist in (ax.lines, ax.patches):
            if not artist == []:
                x.append(np.concatenate([a.get_xdata() for a in artist]))
                y.append(np.concatenate([a.get_ydata() for a in artist]))
        # TODO: get scatter data from ax.collections
        return (np.concatenate(x), np.concatenate(y))

    def _set_frame_position(self, positions):
        """Set positions where frame will be drawn.

        `positions` is a list of strings of frame positions to plot.
            ['left', 'right', 'top', 'bottom' | 'all']
        """
        self._frame_on = self._frame_dict_from(positions)

    def _get_frame_position(self):
        return [p for p in self._position_list if self._frame_on[p]]

    # xposition tuples turn on frame for (bottom, top)
    _xposition_pairs = {(True, False): 'bottom', (False, True): 'top',
                        (True, True): 'both', (False, False): 'none'}
    def _get_xposition(self, frame_on=None):
        """Returns position that matches `XAxis.set_ticks_position` inputs.

        `frame_on` is a dict that matches frame positions with bools.
        """
        if frame_on is None:
            frame_on = self._frame_on
        return self._xposition_pairs[(frame_on['bottom'], frame_on['top'])]

    # yposition tuples turn on frame for (left, right)
    _yposition_pairs = {(True, False): 'left', (False, True): 'right',
                        (True, True): 'both', (False, False): 'none'}
    def _get_yposition(self, frame_on=None):
        """Returns position that matches `YAxis.set_ticks_position` inputs.

        `frame_on` is a dict that matches frame positions with bools.
        """
        if frame_on is None:
            frame_on = self._frame_on
        return self._yposition_pairs[(frame_on['left'], frame_on['right'])]

    def _frame_dict_from(self, positions):
        """Parse `positions` and return xposition, yposition tuple

        `positions` is a list of strings of frame positions to plot.
            ['left', 'right', 'top', 'bottom' | 'all']
        """
        frame_dict = dict.fromkeys(self._position_list, False)
        
        if 'all' in positions:
            frame_dict = dict.fromkeys(self._position_list, True)
        else:
            for position in positions:
                frame_dict[position] = True
        return frame_dict

    def _set_ticks(self):
        """Overide this method to customize tick positioning."""
        # Draw ticks on axes only where a frame is drawn
        self.axes.xaxis.set_ticks_position(self._get_xposition())
        self.axes.yaxis.set_ticks_position(self._get_yposition())

    _frame_lines = dict(bottom=[(0., 0.), (1., 0.)], top=[(0., 1.), (1., 1.)],
                        left=[(0., 0.), (0., 1.)], right=[(1., 0.), (1., 1.)])
    def _make_frame(self):
        """Get axis frame specified by `self._frame_on`."""
        lines = [self._frame_lines[p] for p in self._position_list
                                      if self._frame_on[p]]
        frame_lines = col.LineCollection(segments=lines,
                                         linewidths=[self.linewidth],
                                         colors=[self.color])
        frame_lines.set_transform(self.axes.transAxes)
        return frame_lines

    def draw(self, renderer):
        if not self.get_visible():
            return
        self._set_ticks()
        frame = self._make_frame()
        frame.draw(renderer)

    positions = property(_get_frame_position, _set_frame_position)


class FrameAxes(maxes.Axes):
    """Override Axes so that `frame` attribute is a custom FrameArtist
    """
    name = 'frameaxes'

    def cla(self):
        """Override default Axes.frame and clear patch edgecolor"""
        if not hasattr(self, '_frame'):
            self._frame = Frame(self)
        super(FrameAxes, self).cla()
        self.patch.set_edgecolor('none')
        # retrieve `_frame` because Axes.cla() overwrites `frame`
        self.frame = self._frame
        # self.frame.set_zorder(10)

    def set_frame(self, frame_artist, **kwargs):
        """Set artist to draw axes frame.

        `frame_artist` should be subclass of `FrameArtist`.
        """
        # save frame artist as `_frame` because Axes.cla() overwrites `frame`
        self.frame = self._frame = frame_artist(self, **kwargs)

    def get_frame(self):
        return self.frame
projections.register_projection(FrameAxes)


if __name__ == '__main__':
    # a quick example on different ways to draw the frame
    from numpy.random import rand
    
    positions_test = (None, ['all'], ['top', 'right'])
    names = ('L frame (default)', 'normal frame', 'top, right')
    sub_pos = (221, 223, 122)

    x = rand(10)
    y = rand(10)
    for positions, name, sub in zip(positions_test, names, sub_pos):
        ax = plt.subplot(sub, projection='frameaxes')
        if positions is not None:
            ax.frame.positions = positions
        ax.plot(x, y, 'o')
        ax.set_ylabel(name)
    fig = plt.gcf()
    # remove background color to make the frames stand out
    fig.set_facecolor('w')
    fig.subplots_adjust(hspace=0.5, wspace=0.5)
    plt.show()




Regards,

Zunbeltz

------------------------------------------------------------------------------
Create and Deploy Rich Internet Apps outside the browser with Adobe(R)AIR(TM)
software. With Adobe AIR, Ajax developers can use existing skills and code to
build responsive, highly engaging applications that combine the power of local
resources and data with the reach of the web. Download the Adobe AIR SDK and
Ajax docs to start building applications today-http://p.sf.net/sfu/adobe-com
_______________________________________________
Matplotlib-users mailing list
Matplotlib-users@lists.sourceforge.net
https://lists.sourceforge.net/lists/listinfo/matplotlib-users

Reply via email to