ó áp7]c@sqdZddlmZmZmZddgZd„Zd d„Zd d d„Z d„Z d „Z d „Z d S( s.Helper functions for graphics with Matplotlib.iÿÿÿÿ(tlrangetrangetlongt create_mpl_axtcreate_mpl_figcCs-yddlj}Wntdƒ‚nX|S(s6This function is not needed outside this utils module.iÿÿÿÿNsMatplotlib is not found.(tmatplotlib.pyplottpyplott ImportError(tplt((s9lib/python2.7/site-packages/statsmodels/graphics/utils.pyt _import_mpls cCsF|dkr3tƒ}|jƒ}|jdƒ}n |j}||fS(sHelper function for when a single plot axis is needed. Parameters ---------- ax : Matplotlib AxesSubplot instance, optional If given, this subplot is used to plot in instead of a new figure being created. Returns ------- fig : Matplotlib figure instance If `ax` is None, the created figure. Otherwise the figure to which `ax` is connected. ax : Matplotlib AxesSubplot instance The created axis if `ax` is None, otherwise the axis that was passed in. Notes ----- This function imports `matplotlib.pyplot`, which should only be done to create (a) figure(s) with ``plt.figure``. All other functionality exposed by the pyplot module can and should be imported directly from its Matplotlib module. See Also -------- create_mpl_fig Examples -------- A plotting function has a keyword ``ax=None``. Then calls: >>> from statsmodels.graphics import utils >>> fig, ax = utils.create_mpl_ax(ax) ioN(tNoneR tfiguret add_subplot(taxRtfig((s9lib/python2.7/site-packages/statsmodels/graphics/utils.pyRs %    cCs.|dkr*tƒ}|jd|ƒ}n|S(sHelper function for when multiple plot axes are needed. Those axes should be created in the functions they are used in, with ``fig.add_subplot()``. Parameters ---------- fig : Matplotlib figure instance, optional If given, this figure is simply returned. Otherwise a new figure is created. Returns ------- fig : Matplotlib figure instance If `fig` is None, the created figure. Otherwise the input `fig` is returned. See Also -------- create_mpl_ax tfigsizeN(R R R (RRR((s9lib/python2.7/site-packages/statsmodels/graphics/utils.pyR@s  cCsÖ|dkr%t|jjdƒ}nt|ttfƒrP|j|}|}n|t|tt fƒr´g}g}xX|D]5}t ||ƒ\}}|j |ƒ|j |ƒqxWn|}|jj |ƒ}||fS(sr Give a name or an integer and return the name and integer location of the column in a design matrix. iN( R Rtexogtshapet isinstancetintRt exog_namesttupletlisttmaybe_name_or_idxtappendtindex(tidxtmodelt exog_nametexog_idxtitemtexog_name_itemt exog_idx_item((s9lib/python2.7/site-packages/statsmodels/graphics/utils.pyR^s     cCs¸t|ddƒ}|s-t|ddƒ}n|s¨t|ddgƒ}t|ƒdkr`dn |jd}gt|ƒD] }d^qz}|dkr´|d}q´n |jƒ}|S(s• Input can be an array or pandas-like. Will handle 1d array-like but not 2d. Returns a str for 1d data or a list of strings for 2d data. tnametcolumnsRisX%diN(tgetattrR tlenRRttolist(tseries_or_dataframetnamesRtnvarst_((s9lib/python2.7/site-packages/statsmodels/graphics/utils.pytget_data_namesws%  c KsXxQ|D]I}||}||} ||} |j|| d| ddd||qW|S(s\ Annotate Axes with labels, points, offset_points according to the given index. txytextt textcoordss offset pointstsize(tannotate( Rtlabelstpointst offset_pointsR-R tkwargstitlabeltpointtoffset((s9lib/python2.7/site-packages/statsmodels/graphics/utils.pyt annotate_axesŠs    N( t__doc__tstatsmodels.compat.pythonRRRt__all__R R RRRR*R7(((s9lib/python2.7/site-packages/statsmodels/graphics/utils.pyts  /