W[c @ s"dZddlmZmZddlZddlZddlZddlm Z ddl Z ddl Z ddljZddljZddlmZmZddlmZddlmZee jd kZee jd kZd d d dddgZdZdZ dZ!ddZ"dZ#dZ$ddddZ&dZ'dde(e(e)e)de)dZ*dZ+dZ,ddZ-dddZ.d Z/d!Z0d"Z1dd#Z2e(dd$Z3d%Z4d&Z5dd'Z6d(Z7d)Z8d*Z9dd+d,Z:dS(-s)Small plotting-related utility functions.i(tprint_functiontdivisionN(tstatsi(turlopent urlretrieve(t HTTPException(t LooseVersions0.15s1.5.0t desaturatetsaturatetset_hls_valuestdespinetget_dataset_namest load_datasetcC s|tj|S(s Helper method for removing NA values from array-like. Parameters ---------- arr : array-like The array-like from which to remove NA values. Returns ------- clean_arr : array-like The original array with NA values removed. (tpdtnotnull(tarr((s,lib/python2.7/site-packages/seaborn/utils.pyt remove_nascO s9y|j||SWntk r4|j||SXdS(s=Wrapper to handle different pandas sorting API pre/post 0.17.N(t sort_valuestAttributeErrortsort(tdftargstkwargs((s,lib/python2.7/site-packages/seaborn/utils.pytsort_df*s c C stj|jdd}tj|}g}xZttj|D]C\}\}}||}||}||}|j||gqFWtj|j}|S(sConvert intervals to error arguments relative to plot heights. Parameters ---------- cis: 2 x n sequence sequence of confidence interval limits heights : n sequence sequence of plot heights Returns ------- errsize : 2 x n array sequence of error size relative to height values in correct format as argument for plt.bar ii( tnpt atleast_2dtreshapet atleast_1dt enumeratet transposetappendtasarraytT( tcistheightsterrsizetitlowthighthtelowtehigh((s,lib/python2.7/site-packages/seaborn/utils.pyt ci_to_errsize2s(   i cC sKtj||\}}||j}|d|d}|d ||fS(sKReturn arguments to plt.bar for pmf-like histogram of an array. Parameters ---------- a: array-like array to make histogram of bins: int number of bins Returns ------- x: array left x position of bars h: array height of bars w: float width of bars iii(Rt histogramtsum(tatbinstntxR'tw((s,lib/python2.7/site-packages/seaborn/utils.pytpmf_histPscC sxd|kodkns+tdntjj|}tj|\}}}||9}tj|||}|S(slDecrease the saturation channel of a color by some percent. Parameters ---------- color : matplotlib color hex, rgb-tuple, or html color name prop : float saturation channel of color will be multiplied by this value Returns ------- new_color : rgb tuple desaturated color code in RGB tuple representation iisprop must be between 0 and 1(t ValueErrortmplcoltcolorConvertertto_rgbtcolorsyst rgb_to_hlst hls_to_rgb(tcolortproptrgbR'tltst new_color((s,lib/python2.7/site-packages/seaborn/utils.pyRjs cC st|ddS(s Return a fully saturated color with the same hue. Parameters ---------- color : matplotlib color hex, rgb-tuple, or html color name Returns ------- new_color : rgb tuple saturated color code in RGB tuple representation R>i(R (R:((s,lib/python2.7/site-packages/seaborn/utils.pyRscC sytjj|}ttj|}x<t|||gD]%\}}|dk r=|||R<tvalsR$tval((s,lib/python2.7/site-packages/seaborn/utils.pyR s" cK s0tj}|j|||j||dS(sGrab current axis and label it.N(tplttgcat set_xlabelt set_ylabel(txlabeltylabelRtax((s,lib/python2.7/site-packages/seaborn/utils.pytaxlabels cC s|d kr*|d kr*tjj}n0|d k rB|j}n|d k rZ|g}nx|D]w} xddddgD]} t|  } | j| j| |d k rz| rzy|j| d} Wntk r|} nXt | j| d| fqzqzW|r| rt d| j j D} t d| j j D}| j jdx| j j D]}| |_qgWx | j j D]}||_qWn|r5| r5t d | jj D} t d | jj D}| jjdx| jj D]}| |_qWx | jj D]}||_qWn|ra| j}|jrtj|t| jk|d}tj|t| jk|d }| jdj||| jdj|||j||k}|j||k}| j|n| j}|jrtj|t| jk|d}tj|t| jk|d }| jdj||| jdj|||j||k}|j||k}| j|qqaqaWd S( sRemove the top and right spines from plot(s). fig : matplotlib figure, optional Figure to despine all axes of, default uses current figure. ax : matplotlib axes, optional Specific axes object to despine. top, right, left, bottom : boolean, optional If True, remove that spine. offset : int or dict, optional Absolute distance, in points, spines should be moved away from the axes (negative values move spines inward). A single value applies to all spines; a dict can be used to set offset values per side. trim : bool, optional If True, limit spines to the smallest and largest major tick on each non-despined axis. Returns ------- None ttoptrighttlefttbottomitoutwardcs s|]}|jVqdS(N(ttick1On(t.0tt((s,lib/python2.7/site-packages/seaborn/utils.pys scs s|]}|jVqdS(N(RQ(RRRS((s,lib/python2.7/site-packages/seaborn/utils.pys scs s|]}|jVqdS(N(RQ(RRRS((s,lib/python2.7/site-packages/seaborn/utils.pys scs s|]}|jVqdS(N(RQ(RRRS((s,lib/python2.7/site-packages/seaborn/utils.pys siN(RARDtgcftaxestlocalstspinest set_visibletgetRt_set_spine_positiontanytyaxist majorTickst minorTickstset_ticks_positionttick2Ontxaxist get_xtickstsizeRtcompresstmintget_xlimtmaxt set_boundst set_xtickst get_ytickstget_ylimt set_yticks(tfigRJRLRMRNROtoffsetttrimRUtax_itsidet is_visibleRCtmaj_ontmin_onRStxtickst firstticktlastticktnewtickstyticks((s,lib/python2.7/site-packages/seaborn/utils.pyR sn       !            cC sV|j}|dk r-|j}|j|_n|j||dk rR||_ndS(sL Set the spine's position without resetting an associated axis. As of matplotlib v. 1.0.0, if a spine has an associated axis, then spine.set_position() calls axis.cla(), which resets locators, formatters, etc. We temporarily replace that call with axis.reset_ticks(), which is sufficient for our purposes. N(taxisRAtclat reset_tickst set_position(tspinetpositionRzR{((s,lib/python2.7/site-packages/seaborn/utils.pyRZs     cC sUt|j|||d}t|j|||d}tj|||S(s0Establish support for a kernel density estimate.ii(RgReRtlinspace(tdatatbwtgridsizetcuttclipt support_mint support_max((s,lib/python2.7/site-packages/seaborn/utils.pyt _kde_support,s!!cC sg}yt|}Wn tk r8|g}d}nXxit|D][\}}|dkrytj|j|}ntjtj|||}|j |qFWtj |}|s|j }n|S(sLike scoreatpercentile but can take and return array of percentiles. Parameters ---------- a : array data pcts : sequence of percentile values percentile or percentiles to find score at axis : int or None if not None, computes scores over this axis Returns ------- scores: array array of scores at requested percentiles first dimension is length of object passed to ``pcts`` iN( tlent TypeErrorRRARtscoreatpercentiletravelRtapply_along_axisRRtsqueeze(R-tpctsRztscoresR/R$tptscore((s,lib/python2.7/site-packages/seaborn/utils.pyt percentiles3s    i_cC s,d|dd|df}t|||S(s2Return a percentile range from an array of values.i2i(R(R-twhichRzR((s,lib/python2.7/site-packages/seaborn/utils.pytciXscC sD|dkrdS|dkr dS|dkr0dS|dkr@dSd S( s?Return a R-style significance string corresponding to p values.gMbP?s***g{Gz?s**g?t*g?t.t((R((s,lib/python2.7/site-packages/seaborn/utils.pyt sig_stars^s    cC s;tj|}tj|d}tj|d}||S(s*Calculate the IQR for an array of numbers.iiK(RRRR(R-tq1tq3((s,lib/python2.7/site-packages/seaborn/utils.pytiqrkscC ssddlm}td}||}g|jdidd6D]-}|jjdrB|jjdd^qBS( s?Report available example datasets, useful for reporting issues.i(t BeautifulSoups(https://github.com/mwaskom/seaborn-data/R-sjs-navigation-opentclasss.csvR(tbs4RRtfind_allttexttendswithtreplace(Rthttptgh_listR=((s,lib/python2.7/site-packages/seaborn/utils.pyR ss   cC sk|dkr3tjjdtjjdd}ntjj|}tjj|sgtj|n|S(sCReturn the path of the seaborn data directory. This is used by the ``load_dataset`` function. If the ``data_home`` argument is not specified, the default location is ``~/seaborn-data``. Alternatively, a different default location can be specified using the environment variable ``SEABORN_DATA``. t SEABORN_DATAt~s seaborn-dataN( RAtostenvironRYtpathtjoint expandusertexiststmakedirs(t data_home((s,lib/python2.7/site-packages/seaborn/utils.pyt get_data_homes  cK sDd}|j|}|rmtjjt|tjj|}tjj|sdt||n|}ntj ||}|j dj j r|j d }nt s|S|dkrGtj|dddddg|dddldkr%fdSfd}|S(s Decorator that will skip a test if `url` is unreachable. Parameters ---------- t : function, optional url : str, optional iNc st|dS(Nturl(t_network(R0(R(s,lib/python2.7/site-packages/seaborn/utils.pytsc sQyt}Wn#ttfk r5jnX|j||SdS(N(RtIOErrorRtSkipTesttclose(RRtf(tnoseRSR(s,lib/python2.7/site-packages/seaborn/utils.pytwrappers  (RRA(RSRR((RRSRs,lib/python2.7/site-packages/seaborn/utils.pyRts    (;t__doc__t __future__RRR7RtnumpyRtscipyRtpandasR t matplotlibRtmatplotlib.colorsRR4tmatplotlib.pyplottpyplotRDt!external.six.moves.urllib.requestRRtexternal.six.moves.http_clientRtdistutils.versionRt __version__RRt__all__RRR*R2RRRAR RKtTrueRR RZRRRRRR RR RRRRRRR(((s,lib/python2.7/site-packages/seaborn/utils.pytsT           #  Y   %  :   '   2