o þ0Gfã@s2ddlmZddlZddd„Z   d dd„ZdS) éNÚfactorc Cs@t ¡}|dur t|ƒn|}|}|d|…}| d¡}| t |¡|d|…d¡|jddt |  ¡¡}|d|d}|dt | |g¡7}|  t |¡¡|  |¡t |  ¡¡} d} | d| d}| | t | |g¡7} |  | ¡| d ¡| d ¡| |¡||} t | ¡} | d ¡}| t |¡| d|…d¡| t |¡| d|…d ¡|jddt |  ¡¡}|d|d}|dt | |g¡7}|  t |¡¡|  |¡t |  ¡¡} d} | d| d}| | t | |g¡7} |  | ¡| d ¡| d¡| |¡|jddgdd| ¡|S)a  Plot of the ordered eigenvalues and variance explained for the loadings Parameters ---------- eigenvals : array_like The eigenvalues total_var : float the total variance (for plotting percent variance explained) ncomp : int, optional Number of factors to include in the plot. If None, will included the same as the number of maximum possible loadings x_label : str label of x-axis Returns ------- Figure Handle to the figure. Néyzb-oT)Ztightérg{®Gáz”?z Scree PlotZ Eigenvalueézzg--ozVariance ExplainedZ ProportionZ Cumulativeé)Úloc)ÚpltÚfigureÚlenÚ add_subplotÚplotÚnpZarangeZ autoscaleÚarrayZget_xlimZ set_xticksÚset_xlimZget_ylimÚset_ylimÚ set_titleÚ set_ylabelÚ set_xlabelZcumsumZlegendÚ tight_layout) Z eigenvalsZ total_varZncompZx_labelÚfigZvalsÚaxÚxlimÚspÚylimZscaleZ per_varianceZcumper_variance©rú>lib/python3.10/site-packages/statsmodels/multivariate/plots.pyÚ plot_screesT                rúFactor patternscCsø|j\}}|dur#g}t|ƒD]}t|d|ƒD] } | || g¡qq|dur0dd„t|ƒDƒ}|dur=dd„t|ƒDƒ}g} |D]¸} | d}| d} tjdd} |  | ¡|  d ¡} t|jdƒD]}tj|||f||| f||d d qb|  |dd…|f|dd…| fd ¡|  |¡|dur³d   ||||¡}d   || || ¡}|   |¡|   |¡n|   ||¡|   || ¡d}t   | |g¡}t   | |g¡}|  |ddgd¡|  ddg|d¡|  dd¡|  |¡|  |¡|  ¡qA| S)a¬ Plot factor loadings in 2-d plots Parameters ---------- loadings : array like Each column is a component (or factor) col_names : a list of strings column names of `loadings` row_names : a list of strings row names of `loadings` loading_pairs : None or a list of tuples Specify plots. Each tuple (i, j) represent one figure, i and j is the loading number for x-axis and y-axis, respectively. If `None`, all combinations of the loadings will be plotted. percent_variance : array_like The percent variance explained by each factor. Returns ------- figs : a list of figure handles NrcSóg|]}d|‘qS)z factor %dr©Ú.0ÚirrrÚ lóz!plot_loadings..cSr)zvar %drrrrrr"nr#r)ér$)Zfigsizeéoé )ZfontsizeZboz {} ({:.1f}%)gÍÌÌÌÌÌð?zk--ZequalZdatalim)ÚshapeÚrangeÚappendrr r Útextr rÚformatrrr rZ set_aspectrrr)ZloadingsZ col_namesZ row_namesZ loading_pairsZpercent_varianceÚtitleZk_varZn_factorr!ÚjZfigsÚitemrrÚkZx_strZy_strÚvrrrrrÚ plot_loadingsLsR  ÿ   ÿ&       r1)Nr)NNNNr)Zmatplotlib.pyplotZpyplotrZnumpyr rr1rrrrÚs  Gþ