M/PhFdZddlZddlmZddlmZdgZd dZ d d Z dS)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_ellipseffffff?c fddlm}tj|\}}|dtj|dz }tj|d|dz } d| ztjz } dtj|tj |dzz}||dd|d|dd| zd|d } | |j | d || dS) z%Support function for scatter_ellipse.r)EllipserNnoneg?)angle facecolor edgecolorlw?)matplotlib.patchesr nplinalgeighnormarctanpisqrtrchi2ppf set_clip_boxbbox set_alpha add_artist) meancovaxlevelcolorr vwur ells _/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/statsmodels/graphics/plot_grids.py _make_ellipser*s****** 9>>#  DAq !ry~~ad###A Iad1Q4i E %K"% E BGA ua000 1 11A '$rr(AaD!A$cEkV!   CRWMM#MM#?Fc  tj|}ddlm}t j|}|jd} |dt| D}tdddd } |r| |td } |r| || d} t j |d } td| D]L}d}t|D]6}| | dz | dz |dz | dz z|zdz}| d }|j||j|t j||g}|j|dd|fji| t j|r|g}|D]+}t+| || |dddf|f|fd |i| ,|r&|||d|||sF|s|gnI|j|dn||||r|||nE|s|gn-|j|dt j |d }||dddf|f}|!}|"}|dd|d|dz zz}|ddks|ddkr0| |d|kr|dd|d|dz zz}n|dd|d|dz zz }|dd|d|dz zz}|#||d|dz8N|j$D]}|r-|j|d|r-|j|d|S)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 rNrcg|]}d|zS)zvar%d).0is r) z#scatter_ellipse..as666AGaK666r+r .kr)lsmarkerr$alpha)r$)rowvarz% 3.1fr#-g?)rrgпg?g?g?z $\rho=%0.2f$)%rcreate_mpl_figmatplotlib.tickertickerr asanyarrayshaperangedictupdater r! 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_ticksfigmtickernvars plot_kwds_ ell_kwds_dmeandcovr1ax_lastjr" formatteridxr7dcorrdcxlimylimyrangeqytxts r)rr(sd  s # #C'''''' =  D JqME66u6663cBBBJ%)$$$#I#""" IIaLLE 6$q ! ! !D 1e__;6;6q8 68 6Aq%'AaC%'?13DQ3FGGB 228<>eCjmg6M6M!Wsd1gQ&788!Wsd1gQ&788a3$q'DG"344B GGBORW4 5 5 5 5q8 6th??     + + - - ? H & &w':':1'='= > > >     , , . . ? H & &w':':1'='= > > > Jr+)rN)r,NNNFFN) __doc__numpyrscipyrr__all__r*rr/r+r)rxs    $>BAFIIIIIIr+