ó áp7]c@sFddljZddlZddd„Zdddddd„ZdS(iÿÿÿÿNtfactorc Csætjƒ}|dkr$t|ƒn|}|}|| }|jdƒ}|jtj|ƒ|| dƒ|jdt ƒtj |j ƒƒ}|d|d}|dtj | |gƒ7}|j tj|ƒƒ|j |ƒtj |jƒƒ} d} | d| d}| | tj | |gƒ7} |j| ƒ|jdƒ|jdƒ|j|ƒ||} tj| ƒ} |jd ƒ}|jtj|ƒ| | dƒ|jtj|ƒ| | d ƒ|jdt ƒtj |j ƒƒ}|d|d}|dtj | |gƒ7}|j tj|ƒƒ|j |ƒtj |jƒƒ} d} | d| d}| | tj | |gƒ7} |j| ƒ|jd ƒ|jd ƒ|j|ƒ|jd d gddƒ|jƒ|S(s 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 : string label of x-axis Returns ------- fig : figure Handle to the figure iysb-ottightiig{®Gáz”?s Scree Plott Eigenvalueizsg--osVariance Explainedt Proportiont CumulativetlociN(tplttfiguretNonetlent add_subplottplottnptaranget autoscaletTruetarraytget_xlimt set_xtickstset_xlimtget_ylimtset_ylimt set_titlet set_ylabelt set_xlabeltcumsumtlegendt tight_layout( t eigenvalst total_vartncomptx_labeltfigtvalstaxtxlimtsptylimtscalet per_variancetcumper_variance((s=lib/python2.7/site-packages/statsmodels/multivariate/plots.pyt plot_screesT                 sFactor patternscCsÏ|j\}}|d krlg}xHt|ƒD]7}x.t|d|ƒD]} |j|| gƒqHWq.Wn|d kržgt|ƒD]}d|^q…}n|d krÐgt|ƒD]}d|^q·}ng} xò|D]ê} | d}| d} tjddƒ} | j| ƒ| jdƒ} xLt|jdƒD]7}tj|||f||| f||dd ƒq9W| j|d d …|f|d d …| fd ƒ| j |ƒ|d k r d ||||f}d || || f}| j |ƒ| j |ƒn"| j ||ƒ| j || ƒd }t j | |gƒ}t j | |gƒ}| j|ddgdƒ| jddg|dƒ| jddƒ| j|ƒ| j|ƒ| jƒqÝW| S(s­ 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 is factor %dsvar %ditfigsizeiiotfontsizei Ntbos %s (%.1f%%)gÍÌÌÌÌÌð?sk--tequaltdatalim(ii(tshapeRtrangetappendRRR ttextR RRRR Rt set_aspectRRR(tloadingst col_namest row_namest loading_pairstpercent_variancettitletk_vartn_factortitjtfigstitemR R"tktx_strty_strtvR#R%((s=lib/python2.7/site-packages/statsmodels/multivariate/plots.pyt plot_loadingsLsN  & &     3     (tmatplotlib.pyplottpyplotRtnumpyR RR)RD(((s=lib/python2.7/site-packages/statsmodels/multivariate/plots.pyts  G