>hSHSrSSKJr SS/rSrS SjrS SjrS rS rS r g)z.Helper functions for graphics with Matplotlib.)lrange create_mpl_axcreate_mpl_figc4SSKJn U$! [S5e=f)z6This function is not needed outside this utils module.rNzMatplotlib is not found.)matplotlib.pyplotpyplot ImportError)plts mC:\Users\julio\OneDrive\Documentos\Trabajo\Ideas Frescas\venv\Lib\site-packages\statsmodels/graphics/utils.py _import_mplr s$6' J6455s NcUc.[5nUR5nURS5nX 4$URnX 4$)aYHelper 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 rrsEH zmjjl __S ! 7Nii 7Nc>Uc[5nURUS9nU$)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) JrcUc"[URRS5n[U[5(aUR UnUnX#4$[U[ [45(a?/n/nUH2n[XA5upVURU5 URU5 M4 X#4$UnUR RU5nX#4$)zf 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)    C% ' ' D,=d,J )N   ^ , OOM *   ##))#.  rc,[USS5nU(d [USS5nU(dV[USS/5n[U5S:XaSOURSn[U5Vs/sHnSPM nnUS:XaUSnU$UR 5nU$s snf)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 ^UH&nXnX'n X7n UR"X4U SUS.UD6 M( U$)zP Annotate Axes with labels, points, offset_points according to the given index. z offset points)xytext textcoordssize)annotate) r!labelspoints offset_pointsr8rkwargsilabelpointoffsets r annotate_axesrBsM   ! E )O )!' )  Ir)N)NN) __doc__statsmodels.compat.pythonr__all__r rrrr4rBrr rGs54, , -+\:2& r