'YHhDdZddlZddlmZddlmZdgZd dZ d dZ y) acreate 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)utilsscatter_ellipsec .ddlm}tjj |\}}|dtjj |dz }tj |d|dz } d| ztjz } dtj|tjj|dzz}||dd|d|dd| zd|d } | j|j| jd |j| y) z%Support function for scatter_ellipse.r)EllipserNnoneg?)angle facecolor edgecolorlw?)matplotlib.patchesrnplinalgeighnormarctanpisqrtrchi2ppf set_clip_boxbbox set_alpha add_artist) meancovaxlevelcolorrvwur ells X/var/www/html/planif/env/lib/python3.12/site-packages/statsmodels/graphics/plot_grids.py _make_ellipser)s* 99>># DAq !ryy~~ad##A IIad1Q4i E %K"%% E BGGA ua00 11A $r(AaD!A$cEkV! CRWWMM#MM#c tj|}ddlm}t j |}|j d} |t| D cgc]} d| z }} tdddd } |r| j|td } |r| j||jd} t j|d }td| D]} d}t| D]}|j| dz | dz | dz | dz z|zdz}|jd }|jj||j j|t j"|| g}|j$|dd|fj&i| t j(|r|g}|D]!}t+| |||dddf|f|fd |i| #|r|j-|| d|||j/j1s?|s|j3gn?|jj5|j7dn|j9|| |j/j;r|j=||n>|s|j?gn*|j j5|j7dt j@|d }||dddf|f}|jC}|jE}|dd|d|dz zz}|ddks|ddkr | |d|kDr|dd|d|dz zz}n|dd|d|dz zz }|dd|d|dz zz}|jG||d|dz|jHD]}|j/j;r*|j j5|j7d|j/j1sj|jj5|j7d|Scc} w)aCreate 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 rNrzvar%dr .kr)lsmarkerr#alpha)r#)rowvarz% 3.1fr"-g?)rrgпg?g?g?z $\rho=%0.2f$)%rcreate_mpl_figmatplotlib.tickertickerr asanyarrayshaperangedictupdaterr add_subplotFormatStrFormatteryaxisset_major_formatterxaxisarrayplotTisscalarr) set_titleget_subplotspec is_first_col set_yticksset_major_locator MaxNLocator set_ylabel is_last_row set_xlabel set_xtickscorrcoefget_xlimget_ylimtextaxes)datar"varnamesell_kwds plot_kwds add_titles keep_ticksfigmtickernvarsi plot_kwds_ ell_kwds_dmeandcovax_lastjr! formatteridxr0dcorrdcxlimylimyrangeqytxts r(rr(sd   s #C' == D JJqME).u6AGaK663cBJ)$#I" IIaLE 66$q !D 1e_;6q8 6Aq%'AaC%'?13DQ3FGB 228eCjmg6M!Wsd1gQ&788!Wsd1gQ&788a3$q'DG"344B GGBORW4 5q8 6;6zhh?    + + - HH & &w':':1'= >    , , . HH & &w':':1'= > ? Ja7s Q)gffffff?N)g?NNNFFN) __doc__numpyrscipyrr__all__r)rr*r(rss4   $>BAFIr*