U Kv fã@s,ddlmZddlZddd„Zd dd„ZdS) éNÚfactorc Cs@t ¡}|dkrt|ƒ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ÚfigÚvalsÚaxÚxlimÚspÚylimZscaleZ per_varianceZcumper_variance©rúB/tmp/pip-unpacked-wheel-2v6byqio/statsmodels/multivariate/plots.pyÚ plot_screesT                rúFactor patternscCsü|j\}}|dkrFg}t|ƒD]&}t|d|ƒD]} | || g¡q0q|dkr`dd„t|ƒDƒ}|dkrzdd„t|ƒDƒ}g} |D]r} | d}| d} tjdd} |  | ¡|  d ¡} t|jdƒD]*}tj|||f||| f||d d qÆ|  |dd…|f|dd…| fd ¡|  |¡|dk rjd ||||f}d || || f}|   |¡|   |¡n|   ||¡|   || ¡d}t   | |g¡}t   | |g¡}|  |ddgd¡|  ddg|d¡|   dd¡|  |¡|  |¡|  ¡q‚| 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 NrcSsg|] }d|‘qS)z factor %dr©Ú.0ÚirrrÚ lsz!plot_loadings..cSsg|] }d|‘qS)zvar %drrrrrr"nsr)ér#)Zfigsizeéoé )ZfontsizeZboz %s (%.1f%%)gÍÌÌÌÌÌð?zk--ÚequalZdatalim)ÚshapeÚrangeÚappendrr r Útextr rrrr rZ set_aspectrrr)ZloadingsZ col_namesZ row_namesZ loading_pairsZpercent_varianceÚtitleZk_varZn_factorr!ÚjZfigsÚitemrrÚkZx_strZy_strÚvrrrrrÚ plot_loadingsLsR      ÿ&        r0)Nr)NNNNr)Zmatplotlib.pyplotZpyplotrZnumpyr rr0rrrrÚs  Gþ