!gDdZddlmZddgZdZd dZd dZd Zd Zd Z y)z.Helper functions for graphics with Matplotlib.)lrange create_mpl_axcreate_mpl_figc4 ddlm}|S#tdxYw)z6This function is not needed outside this utils module.rNzMatplotlib is not found.)matplotlib.pyplotpyplot ImportError)plts W/var/www/dash_apps/app1/venv/lib/python3.12/site-packages/statsmodels/graphics/utils.py _import_mplr s$6' J6455s Nc|/t}|j}|jd}||fS|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) o)r figure add_subplot)axr figs r rrsIH zmjjl __S ! 7Nii 7NcB|t}|j|}|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 )figsize)r r)rrr s r rr?s%, {mjjj) Jrc|"t|jjd}t|tr|j |}|}||fSt|t tfr@g}g}|D]3}t||\}}|j||j|5||fS|}|j j|}||fS)zr Give a name or an integer and return the name and integer location of the column in a design matrix. ) rexogshape isinstanceint exog_namestuplelistmaybe_name_or_idxappendindex)idxmodel exog_nameexog_idxitemexog_name_item exog_idx_items r rr\s  {UZZ%%a()#s$$S)  h  C% '  +D,=dE,J )NM   ^ , OOM * + h  ##))#. h rct|dd}|s t|dd}|sQt|ddg}t|dk(rdn|jd}t|Dcgc]}d}}|dk(r|d}|S|j }|Scc}w)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. nameNcolumnsrrzX%dr)getattrlenrrangetolist)series_or_dataframenamesrnvars_s r get_data_namesr4us ' 6E +Y= +Wqc:Z1_*=*C*CA*F %e -1-- A:!HE L  L .s Bc b|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_pointsr8rkwargsilabelpointoffsets r annotate_axesrBs\ )q q q! E5 )O )!' ) ) Ir)N)NN) __doc__statsmodels.compat.pythonr__all__r rrrr4rBrr rGs54, , -+\:2& r