B ¤ŠãZöã@sNdZddlmZddlZddlmZddlmZdgZ dd d „Z dd d„Z dS)a®create scatterplot with confidence ellipsis Author: Josef Perktold License: BSD-3 TODO: update script to use sharex, sharey, and visible=False see http://www.scipy.org/Cookbook/Matplotlib/Multiple_Subplots_with_One_Axis_Label for sharex I need to have the ax of the last_row when editing the earlier rows. Or you axes_grid1, imagegrid http://matplotlib.sourceforge.net/mpl_toolkits/axes_grid/users/overview.html é)ÚrangeN)Ústatsé)ÚutilsÚscatter_ellipseçffffffî?c CsÂddlm}tj |¡\}}|dtj |d¡}t |d|d¡} d| tj} dt |t j   |d¡¡}||dd…|d|dd| d|dd } |   |j ¡|  d ¡| | ¡dS) z%Support function for scatter_ellipse.r)ÚEllipseré´éNÚnonegø?)Z facecolorZ edgecolorZlwgà?)Zmatplotlib.patchesrÚnpZlinalgZeighZnormZarctanZpiZsqrtrZchi2ZppfZ set_clip_boxZbboxZ set_alphaZ add_artist) ÚmeanÚcovÚaxÚlevelÚcolorrÚvÚwÚuZangleZell©rú>lib/python3.7/site-packages/statsmodels/graphics/plot_grids.pyÚ _make_ellipses    rçÍÌÌÌÌÌì?Fc Cs`t |¡}ddlm}t |¡}|jd} |dkrDdd„t| ƒDƒ}tdddd d } |rb|   |¡tdd } |rz|   |¡|  d¡} tj |dd } x|td| ƒD]l}d}x`t|ƒD]R}|  | d| d|d| d|d¡}|  d ¡}|j |¡|j |¡t ||g¡}|j|dd…|fj| Žt |¡rB|g}x>|D]6}t| || |dd…df|f|fd|i| —ŽqHW|r¢| d||||f¡| ¡sÒ|s¾| g¡n|j | d¡¡n| ||¡| ¡rú| ||¡n$|s | g¡n|j | d¡¡tj|dd }||dd…df|f}|  ¡}| !¡}|dd|d|d}|ddksž|ddkr¼| |d|kr¼|dd|d|d}n|dd|d|d}|dd|d|d}| "||d|d¡q¶Wq WxH|j#D]>}| ¡r:|j | d¡¡| ¡r|j | d¡¡qW|S)aCCreate a grid of scatter plots with confidence ellipses. ell_kwds, plot_kdes not used yet looks ok with 5 or 6 variables, too crowded with 8, too empty with 1 Parameters ---------- data : array_like Input data. level : scalar, optional Default is 0.9. varnames : list of str, optional Variable names. Used for y-axis labels, and if `add_titles` is True also for titles. If not given, integers 1..data.shape[1] are used. ell_kwds : dict, optional UNUSED plot_kwds : dict, optional UNUSED add_titles : bool, optional Whether or not to add titles to each subplot. Default is False. Titles are constructed from `varnames`. keep_ticks : bool, optional If False (default), remove all axis ticks. 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 `fig` itself. rNrcSsg|] }d|‘qS)zvar%dr)Ú.0Úirrrú Vsz#scatter_ellipse..r Ú.Úkgà?)ZlsZmarkerrÚalpha)r)Zrowvarz% 3.1frz%s-%ségš™™™™™Ù?)rrgпgÐ?gš™™™™™¹?gš™™™™™É?z $\rho=%0.2f$)$rZcreate_mpl_figZmatplotlib.tickerZtickerr Z asanyarrayÚshaperÚdictÚupdater rZ add_subplotZFormatStrFormatterZyaxisZset_major_formatterZxaxisZarrayZplotÚTZisscalarrZ set_titleZ is_first_colZ set_yticksZset_major_locatorZ MaxNLocatorZ set_ylabelZ is_last_rowZ set_xlabelZ set_xticksZcorrcoefZget_xlimZget_ylimÚtextZaxes)ÚdatarZvarnamesZell_kwdsZ plot_kwdsZ add_titlesZ keep_ticksZfigZmtickerZnvarsZ plot_kwds_Z ell_kwds_ZdmeanZdcovrZax_lastÚjrZ formatterÚidxrZdcorrZdcZxlimZylimZyrangeqZytZxtrrrr*sp%        *     (    .   )rN)rNNNFFN) Ú__doc__Zstatsmodels.compat.pythonrZnumpyr ZscipyrÚrÚ__all__rrrrrrÚ s