U Kv f©ã@sBdZddlZddlmZddlmZdgZd dd „Zdd 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 https://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 éN)Ú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 edgecolorZlwçà?)Zmatplotlib.patchesrÚnpZlinalgZeighZnormZarctanÚpiÚsqrtrZchi2ZppfZ set_clip_boxZbboxZ set_alphaZ add_artist) ÚmeanÚcovÚaxÚlevelÚcolorrÚvÚwÚuZangleZell©rúC/tmp/pip-unpacked-wheel-2v6byqio/statsmodels/graphics/plot_grids.pyÚ _make_ellipses  ý  rçÍÌÌÌÌÌì?Fc CsLt |¡}ddlm}t |¡}|jd} |dkrDdd„t| ƒDƒ}tdddd d } |rb|   |¡tdd } |rz|   |¡|  d¡} tj |dd } td| ƒD]b}d}t|ƒD]N}|  | d| d|d| d|d¡}|  d ¡}|j |¡|j |¡t ||g¡}|j|dd…|fj| Žt |¡r:|g}|D]6}t| || |dd…df|f|fd|i| —Žq>|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®qœ|j#D]>}| ¡r(|j | d¡¡| ¡r|j | d¡¡q|S)aÈCreate 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[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 : 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 `fig` itself. Examples -------- >>> import statsmodels.api as sm >>> import matplotlib.pyplot as plt >>> import numpy as np >>> from statsmodels.graphics.plot_grids import scatter_ellipse >>> data = sm.datasets.statecrime.load_pandas().data >>> fig = plt.figure(figsize=(8,8)) >>> scatter_ellipse(data, varnames=data.columns, fig=fig) >>> plt.show() .. plot:: plots/graphics_plot_grids_scatter_ellipse.py rNrcSsg|] }d|‘qS)zvar%dr)Ú.0ÚirrrÚ asz#scatter_ellipse..r Ú.Úkr )ZlsÚmarkerrÚalpha)r)Zrowvarz% 3.1frz%s-%ségš™™™™™Ù?)rrgпgÐ?gš™™™™™¹?gš™™™™™É?z $\rho=%0.2f$)$rZcreate_mpl_figZmatplotlib.tickerZtickerr Z asanyarrayÚshapeÚrangeÚdictÚupdaterrZ add_subplotZFormatStrFormatterZyaxisZset_major_formatterZxaxisÚarrayZplotÚTZisscalarrÚ set_titleZ is_first_colZ set_yticksZset_major_locatorZ MaxNLocatorZ set_ylabelZ is_last_rowZ set_xlabelZ set_xticksZcorrcoefZget_xlimZget_ylimÚtextZaxes)ÚdatarÚvarnamesZell_kwdsZ plot_kwdsZ add_titlesZ keep_ticksZfigZmtickerZnvarsZ plot_kwds_Z ell_kwds_ZdmeanZdcovrZax_lastÚjrÚ formatterÚidxr!ZdcorrÚdcZxlimZylimZyrangeqZytZxtrrrr(sr2        *    (ÿ     .   )rN)rNNNFFN) Ú__doc__Znumpyr ZscipyrÚrÚ__all__rrrrrrÚs    þ