'YHhK4dZddlZddlmZ ddZ ddZy) zcorrelation plots Author: Josef Perktold License: BSD-3 example for usage with different options in statsmodels/sandbox/examples/thirdparty/ex_ratereturn.py N)utilsc |d}nd}tj|\}}|jd} ||}|d}t|tr|\} } n |rd\} } nd\} } |j ||dd| d| f| | } t jd| d z} t|tr t|dk(r|jgnF|D|j| |j| dd d zd |j|ddd d dt|tr t|dk(r|jgnA|?|j| |j| dd d zd |j|d dd|dk(s|j||r|j| d|j!|j#dd|j#ddd |j%ddddd|S#t&$rY|SwxYw)aJPlot correlation of many variables in a tight color grid. Parameters ---------- dcorr : ndarray Correlation matrix, square 2-D array. xnames : list[str], optional Labels for the horizontal axis. If not given (None), then the matplotlib defaults (integers) are used. If it is an empty list, [], then no ticks and labels are added. ynames : list[str], optional Labels for the vertical axis. Works the same way as `xnames`. If not given, the same names as for `xnames` are re-used. title : str, optional The figure title. If None, the default ('Correlation Matrix') is used. If ``title=''``, then no title is added. normcolor : bool or tuple of scalars, optional If False (default), then the color coding range corresponds to the range of `dcorr`. If True, then the color range is normalized to (-1, 1). If this is a tuple of two numbers, then they define the range for the color bar. ax : AxesSubplot, optional If `ax` is None, then a figure is created. If an axis instance is given, then only the main plot but not the colorbar is created. cmap : str or Matplotlib Colormap instance, optional The colormap for the plot. Can be any valid Matplotlib Colormap instance or name. Returns ------- Figure If `ax` is None, the created figure. Otherwise the figure to which `ax` is connected. Examples -------- >>> import numpy as np >>> import matplotlib.pyplot as plt >>> import statsmodels.graphics.api as smg >>> hie_data = sm.datasets.randhie.load_pandas() >>> corr_matrix = np.corrcoef(hie_data.data.T) >>> smg.plot_corr(corr_matrix, xnames=hie_data.names) >>> plt.show() .. plot:: plots/graphics_correlation_plot_corr.py NTFrzCorrelation Matrix)gg?)NNnearest)cmap interpolationextentvminvmaxg?)minorsmallright)fontsizehorizontalalignment-)rrotationr) use_gridspecr )whichlengthout) directiontopr-wr)r linestylecolorlw)r create_mpl_axshape isinstancetupleimshownparangelistlen set_yticksset_yticklabels set_xticksset_xticklabels set_titlecolorbar tight_layout tick_paramsgridAttributeError)dcorrxnamesynamestitle normcoloraxrcreate_colorbarfignvarsr r aximlabelPoss Y/var/www/html/planif/env/lib/python3.12/site-packages/statsmodels/graphics/correlation.py plot_corrr?sb z!!"%GC KKNE ~ }$)U# d  d d 99UYuQu-Dt  EDyyE"S(H&$CK1$4 b   h hsmC't 4 6$B$<'/6  8&$CK1$4 b   h hsmC't 4 6Gb/6  8 B; U T -NNN+NNUUN;  Gs#!D J   J  sG00 G=<G=c J||}|sdgt|z}t|}|+ttj|t |z } nR|dkrd|}} nHttj |} ttj|t | z }t |t | z d} tj | dz} tj|| | zdz| f}t|D]Q\} } |j| || dz}| |z| dzz |kr|ng}| dz|zdk(r|ng}t| |||| |||S|jdd d d |jgd }|j|jd j d ||S)a Create a grid of correlation plots. The individual correlation plots are assumed to all have the same variables, axis labels can be specified only once. Parameters ---------- dcorrs : list or iterable of ndarrays List of correlation matrices. titles : list[str], optional List of titles for the subplots. By default no title are shown. ncols : int, optional Number of columns in the subplot grid. If not given, the number of columns is determined automatically. normcolor : bool or tuple, optional If False (default), then the color coding range corresponds to the range of `dcorr`. If True, then the color range is normalized to (-1, 1). If this is a tuple of two numbers, then they define the range for the color bar. xnames : list[str], optional Labels for the horizontal axis. If not given (None), then the matplotlib defaults (integers) are used. If it is an empty list, [], then no ticks and labels are added. ynames : list[str], optional Labels for the vertical axis. Works the same way as `xnames`. If not given, the same names as for `xnames` are re-used. fig : Figure, optional If given, this figure is simply returned. Otherwise a new figure is created. cmap : str or Matplotlib Colormap instance, optional The colormap for the plot. Can be any valid Matplotlib Colormap instance or name. Returns ------- Figure If `ax` is None, the created figure. Otherwise the figure to which `ax` is connected. Examples -------- >>> import numpy as np >>> import matplotlib.pyplot as plt >>> import statsmodels.api as sm In this example we just reuse the same correlation matrix several times. Of course in reality one would show a different correlation (measuring a another type of correlation, for example Pearson (linear) and Spearman, Kendall (nonlinear) correlations) for the same variables. >>> hie_data = sm.datasets.randhie.load_pandas() >>> corr_matrix = np.corrcoef(hie_data.data.T) >>> sm.graphics.plot_corr_grid([corr_matrix] * 8, xnames=hie_data.names) >>> plt.show() .. plot:: plots/graphics_correlation_plot_corr_grid.py rrg?)figsize)r4r5r6r7r8r皙?g ףp= ?g?)bottomleftrr)gq= ףp?rDg?g?r)cax)r(intr%ceilfloatsqrtminrcreate_mpl_fig enumerate add_subplotr?subplots_adjustadd_axesr.axesimages)dcorrstitlesncolsr7r4r5r:rn_plotsnrowsaspectvsizeicr8_xnames_ynamesrGs r>plot_corr_gridr_}sx~ c&k!&kG BGGGeEl234 Q;g5E()E%, 678Eu%s +F GGENQ E   sUV^a-?,G HC&!91 __UE1Q3 /!EMQqS1E9&rqSEMQ.&B!GG6!9%"4 9 9sScB ,,. /CLL!##A&CL0 J)NNNFNRdYlBu_r)NNFNNNra)__doc__numpyr%rrr?r_r`r>res3FK&k\MQ/9`r`