M/PhFdZddlmZddgZdZd dZd dZd Zd Zd Z dS)z.Helper functions for graphics with Matplotlib.)lrange create_mpl_axcreate_mpl_figc> ddlm}n#tdxYw|S)z6This function is not needed outside this utils module.rNzMatplotlib is not found.)matplotlib.pyplotpyplot ImportError)plts Z/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/statsmodels/graphics/utils.py _import_mplr s96'''''''64555 Js Nc|8t}|}|d}n|j}||fS)aHelper function for when a single plot axis is needed. Parameters ---------- ax : AxesSubplot, optional If given, this subplot is used to plot in instead of a new figure being created. Returns ------- fig : Figure If `ax` is None, the created figure. Otherwise the figure to which `ax` is connected. ax : AxesSubplot 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) No)r figure add_subplot)axr figs r rrsCH zmmjjll __S ! !i 7NcR|$t}||}|S)aHelper 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 : Figure, optional If given, this figure is simply returned. Otherwise a new figure is created. Returns ------- Figure If `fig` is None, the created figure. Otherwise the input `fig` is returned. See Also -------- create_mpl_ax N)figsize)r r)rrr s r rr?s*, {mmjjj)) Jrc|t|jjd}t|tr|j|}|}nt|t tfrGg}g}|D]?}t||\}}| || |@n|}|j |}||fS)zr Give a name or an integer and return the name and integer location of the column in a design matrix. N) rexogshape isinstanceint exog_namestuplelistmaybe_name_or_idxappendindex)idxmodel exog_nameexog_idxitemexog_name_item exog_idx_items r rr\s  {UZ%a())#s /$S)  C% ' ' /  + +D,=dE,J,J )NM   ^ , , , OOM * * * * +  #))#.. h rc2t|dd}|st|dd}|s\t|ddg}t|dkrdn |jd}dt|D}|dkr|d}n|}|S)z 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. nameNcolumnsrrcg|]}dS)zX%d).0_s r z"get_data_names..s---1---rr)getattrlenrrangetolist)series_or_dataframenamesrnvarss r get_data_namesr8us ' 6 6E >+Y== +Wqc::ZZ1__*=*CA*F--e --- A::!HE  Lrc f|D]-}||}||} ||} |j|| f| d|d|.|S)z\ Annotate Axes with labels, points, offset_points according to the given index. z offset points)xytext textcoordssize)annotate) r!labelspoints offset_pointsr<rkwargsilabelpointoffsets r annotate_axesrFsl ))q q q! E5 )O ) )!' ) ) ) ) Ir)N)NN) __doc__statsmodels.compat.pythonr__all__r rrrr8rFr-rr rJs44,,,,,, , -++++\:2&     r