B A—W[Kã@s’dZddlmZmZddlZddlZddlZddlm Z ddl Z ddl Z ddlmZddlmZddlmZmZddlmZddlmZee jƒd kZee jƒd kZd d d dddgZdd„Zdd„Z dd„Z!dBdd„Z"dd „Z#dd „Z$dCdd „Z%dd„Z&dDd!d„Z'd"d#„Z(d$d%„Z)dEd&d'„Z*dFd)d*„Z+d+d,„Z,d-d.„Z-d/d„Z.dGd0d1„Z/dHd2d„Z0d3d4„Z1d5d6„Z2dId7d8„Z3d9d:„Z4d;d<„Z5d=d>„Z6dJd@dA„Z7dS)Kz)Small plotting-related utility functions.é)Úprint_functionÚdivisionN)Ústatsé)ÚurlopenÚ urlretrieve)Ú HTTPException)Ú LooseVersionz0.15z1.5.0Ú desaturateÚsaturateÚset_hls_valuesÚdespineÚget_dataset_namesÚ load_datasetcCs|t |¡S)a 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. )ÚpdÚnotnull)Zarr©rú,lib/python3.7/site-packages/seaborn/utils.pyÚ remove_nasrcOs.y |j||ŽStk r(|j||ŽSXdS)z=Wrapper to handle different pandas sorting API pre/post 0.17.N)Z sort_valuesÚAttributeErrorÚsort)ÚdfÚargsÚkwargsrrrÚsort_df*s rc Csvt |¡ dd¡}t |¡}g}xDtt |¡ƒD]2\}\}}||}||}||}| ||g¡q0Wt |¡j}|S)a‚Convert 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 ééÿÿÿÿ) ÚnpZ atleast_2dZreshapeZ atleast_1dÚ enumerateZ transposeÚappendÚasarrayÚT) ZcisZheightsZerrsizeÚiZlowZhighÚhZelowZehighrrrÚ ci_to_errsize2s  r$é cCs>t ||¡\}}|| ¡}|d|d}|dd…||fS)aKReturn 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 rrNr)rZ histogramÚsum)ÚaZbinsÚnÚxr#ÚwrrrÚpmf_histPs r+cCsTd|krdksntdƒ‚tj |¡}tj|Ž\}}}||9}t |||¡}|S)alDecrease 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 rrzprop must be between 0 and 1)Ú ValueErrorÚmplcolÚcolorConverterÚto_rgbÚcolorsysÚ rgb_to_hlsÚ hls_to_rgb)ÚcolorZpropÚrgbr#ÚlÚsZ new_colorrrrr js cCs t|ddS)a 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)r6)r )r3rrrr scCsTtj |¡}ttj|Žƒ}x*t|||gƒD]\}}|dk r*|||<q*Wtj|Ž}|S)anIndependently manipulate the h, l, or s channels of a color. Parameters ---------- color : matplotlib color hex, rgb-tuple, or html color name h, l, s : floats between 0 and 1, or None new values for each channel in hls space Returns ------- new_color : rgb tuple new color code in RGB tuple representation N)r-r.r/Úlistr0r1rr2)r3r#r5r6r4Zvalsr"Úvalrrrr žs   cKs(t ¡}|j|f|Ž|j|f|ŽdS)zGrab current axis and label it.N)ÚpltZgcaZ set_xlabelZ set_ylabel)ZxlabelZylabelrÚaxrrrÚaxlabel¹sr;TFc CsÚ|dkr|dkrt ¡j}n|dk r,|j}n|dk r:|g}x˜|D]Ž} xrdD]j} tƒ|  } | j|  | ¡|dk rN| rNy| | d¡} Wntk r¢|} YnXt| j| d| fƒqNW|r4|s4t dd„| j j Dƒƒ} t dd„| j j Dƒƒ}| j   d¡x| j j D] }| |_q Wx| j j D] }||_q$W|r¬|s¬t d d„| jj Dƒƒ} t d d„| jj Dƒƒ}| j  d ¡x| jj D] }| |_q‚Wx| jj D] }||_qœW|rB|  ¡}|jrBt |t|  ¡ƒk|¡d}t |t|  ¡ƒk|¡d }| jd  ||¡| jd  ||¡| ||k¡}| ||k¡}|  |¡|  ¡}|jrBt |t|  ¡ƒk|¡d}t |t|  ¡ƒk|¡d }| jd ||¡| jd ||¡| ||k¡}| ||k¡}|  |¡qBWdS)aìRemove 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 N)ÚtopÚrightÚleftÚbottomrZoutwardcss|] }|jVqdS)N)Útick1On)Ú.0Útrrrú îszdespine..css|] }|jVqdS)N)r@)rArBrrrrCïsr=css|] }|jVqdS)N)r@)rArBrrrrC÷scss|] }|jVqdS)N)r@)rArBrrrrCøsr<rr?r>)r9ZgcfÚaxesÚlocalsZspinesZ set_visibleÚgetrÚ_set_spine_positionÚanyZyaxisZ majorTicksZ minorTicksZset_ticks_positionZtick2OnZxaxisZ get_xticksÚsizerÚcompressÚminZget_xlimÚmaxZ set_boundsZ set_xticksZ get_yticksZget_ylimZ set_yticks)Zfigr:r<r=r>r?ÚoffsetZtrimrDZax_iZsideZ is_visibler8Zmaj_onZmin_onrBZxticksZ firsttickZlasttickZnewticksZyticksrrrr Àsn                  cCs8|j}|dk r|j}|j|_| |¡|dk r4||_dS)aL 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)ÚaxisÚclaZ reset_ticksZ set_position)ZspineZpositionrNrOrrrrGs  rGcCsBt| ¡|||dƒ}t| ¡|||dƒ}t |||¡S)z0Establish support for a kernel density estimate.rr)rLrKrZlinspace)ÚdataZbwZgridsizeZcutZclipZ support_minZ support_maxrrrÚ _kde_support,srQcCs–g}y t|ƒ}Wntk r.|g}d}YnXxJt|ƒD]>\}}|dkr\t | ¡|¡}nt tj|||¡}| |¡q:Wt  |¡}|s’|  ¡}|S)aÆLike 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`` rN) ÚlenÚ TypeErrorrrÚscoreatpercentileZravelrZapply_along_axisrr Zsqueeze)r'ZpctsrNZscoresr(r"ÚpZscorerrrÚ percentiles3s   rVé_cCs$d|dd|df}t|||ƒS)z2Return a percentile range from an array of values.é2r)rV)r'ZwhichrNrUrrrÚciXsrYcCs4|dkr dS|dkrdS|dkr$dS|dkr0dSd S) z?Return a R-style significance string corresponding to p values.gü©ñÒMbP?z***g{®Gáz„?z**gš™™™™™©?Ú*gš™™™™™¹?Ú.Úr)rUrrrÚ sig_stars^sr]cCs*t |¡}t |d¡}t |d¡}||S)z*Calculate the IQR for an array of numbers.ééK)rr rrT)r'Zq1Zq3rrrÚiqrks   r`cCs6ddlm}tdƒ}||ƒ}dd„| dddi¡DƒS) z?Report available example datasets, useful for reporting issues.r)Ú BeautifulSoupz(https://github.com/mwaskom/seaborn-data/cSs&g|]}|j d¡r|j dd¡‘qS)z.csvr\)ÚtextÚendswithÚreplace)rAr5rrrú zsz%get_dataset_names..r'Úclasszjs-navigation-open)Zbs4rarZfind_all)raZhttpZgh_listrrrrss  cCsF|dkr tj dtj dd¡¡}tj |¡}tj |¡sBt |¡|S)aCReturn 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``. NZ SEABORN_DATAú~z seaborn-data)ÚosÚenvironrFÚpathÚjoinÚ expanduserÚexistsÚmakedirs)Ú data_homerrrÚ get_data_homes    rpcKs¢d}| |¡}|rFtj t|ƒtj |¡¡}tj |¡sBt||ƒ|}tj |f|Ž}|j d  ¡  ¡rt|j dd…}t s||S|dkrèt |ddddd g¡|d<t |d d d g¡|d <t |d ddg¡|d <t |dddg¡|d<|dkr t |d|j ¡¡|d<|dkrbt |d dddg¡|d <t |ddddg¡|d<t |dddg¡|d<|d kržt |d!d"d#d$g¡|d!<t |d%td&ƒ¡|d%<|S)'a:Load a dataset from the online repository (requires internet). Parameters ---------- name : str Name of the dataset (`name`.csv on https://github.com/mwaskom/seaborn-data). You can obtain list of available datasets using :func:`get_dataset_names` cache : boolean, optional If True, then cache data locally and use the cache on subsequent calls data_home : string, optional The directory in which to cache data. By default, uses ~/seaborn-data/ kws : dict, optional Passed to pandas.read_csv zDhttps://raw.githubusercontent.com/mwaskom/seaborn-data/master/{}.csvrNZtipsZdayZThurZFriZSatZSunZsexZMaleZFemaleZtimeZLunchZDinnerZsmokerZYesZNoZflightsÚmonthZexercisez1 minz15 minz30 minZkindÚrestZwalkingZrunningZdietzno fatzlow fatZtitanicrfZFirstZSecondZThirdZdeckZABCDEFG)ÚformatrhrjrkrpÚbasenamermrrZread_csvZilocZisnullÚallÚpandas_has_categoricalsZ CategoricalrqÚuniquer7)ÚnameÚcacheroZkwsrjÚ full_pathÚ cache_pathrrrrr“s8       csN|sdSy,dd„|Dƒ‰‡fdd„ˆDƒ}t|ƒdkStk rHdSXdS)záReturn a boolean for whether the list of ticklabels have overlaps. Parameters ---------- labels : list of ticklabels Returns ------- overlap : boolean True if any of the labels overlap. FcSsg|] }| ¡‘qSr)Zget_window_extent)rAr5rrrreÝsz+axis_ticklabels_overlap..csg|]}| ˆ¡‘qSr)Zcount_overlaps)rAÚb)ÚbboxesrrreÞsrN)rLÚ RuntimeError)ÚlabelsZoverlapsr)r}rÚaxis_ticklabels_overlapÍs  r€cCst| ¡ƒt| ¡ƒfS)zôReturn booleans for whether the x and y ticklabels on an Axes overlap. Parameters ---------- ax : matplotlib Axes Returns ------- x_overlap, y_overlap : booleans True when the labels on that axis overlap. )r€Zget_xticklabelsZget_yticklabels)r:rrrÚaxes_ticklabels_overlapås rc Cs¾|dkr¶t|dƒr|j}ny |jj}Wn‚ttfk r¨y | ¡}Wntk rdt |¡}YnXy t |¡  tj ¡t  |¡}Wnt tfk r¢|}YnXYnXt tj|ƒ}t|ƒS)aÔReturn a list of unique data values. Determine an ordered list of levels in ``values``. Parameters ---------- values : list, array, Categorical, or Series Vector of "categorical" values order : list-like, optional Desired order of category levels to override the order determined from the ``values`` object. Returns ------- order : list Ordered list of category levels not including null values. NÚ categories)Úhasattrr‚ÚcatrSrrwrrr ZastypeÚfloatrr,Úfilterrr7)ÚvaluesÚorderrrrÚcategorical_orderös"    r‰cCs<tr2tjd}ydd„|DƒStk r0YnXtjdS)z@Return the list of colors in the current matplotlib color cycle.zaxes.prop_cyclecSsg|] }|d‘qS)r3r)rAr)rrrre%sz#get_color_cycle..zaxes.color_cycle)Ú mpl_ge_150ÚmplZrcParamsÚKeyError)ZcylrrrÚget_color_cycles rcCsrtjj |¡dd…dd…f}t |dk|d|ddd¡}| dd d g¡}y| ¡Stk rl|SXdS) aCalculate the relative luminance of a color according to W3C standards Parameters ---------- color : matplotlib color or sequence of matplotlib colors Hex code, rgb-tuple, or html color name. Returns ------- luminance : float(s) between 0 and 1 Nég#Æ‚¤?g×£p= ×)@g)\Âõ(¬?gáz®Gáð?g333333@g¼–z6Ë?g¥,Cëâæ?g]mÅþ²{²?) r‹Úcolorsr.Z to_rgba_arrayrÚwhereÚdotÚitemr,)r3r4ZlumrrrÚrelative_luminance,s "r“cCs€t|tƒr*y | d¡Stk r(|SXy t|tƒr:|S| ¡ d¡SWn0tk rzt|tƒrn| d¡S| ¡SYnXdS)apReturn a Unicode string representing a Python object. Unicode strings (i.e. type ``unicode`` in Python 2.7 and type ``str`` in Python 3.x) are returned unchanged. Byte strings (i.e. type ``str`` in Python 2.7 and type ``bytes`` in Python 3.x) are returned as UTF-8-encoded strings. For other objects, the method ``__str__()`` is called, and the result is returned as a UTF-8-encoded string. Parameters ---------- obj : object Any Python object Returns ------- s : unicode (Python 2.7) / str (Python 3.x) UTF-8-encoded string representation of ``obj`` zutf-8N)Ú isinstanceÚstrÚdecoderZunicodeÚ__str__Ú NameErrorÚbytes)ÚobjrrrÚto_utf8Bs     r›úhttps://google.comcs0ddl‰ˆdkr‡fdd„S‡‡‡fdd„}|S)z” Decorator that will skip a test if `url` is unreachable. Parameters ---------- t : function, optional url : str, optional rNcs t|ˆdS)N)Úurl)Ú_network)r))rrrÚ€sz_network..c sDy tˆƒ}Wn ttfk r,ˆ ¡‚YnX| ¡ˆ||ŽSdS)N)rÚIOErrorrZSkipTestÚclose)rrÚf)ÚnoserBrrrÚwrapper‚s  z_network..wrapper)r£)rBrr¤r)r£rBrrržts   rž)r%)NNN)NNTTFFNF)N)rWN)N)TN)N)Nrœ)8Ú__doc__Z __future__rrr0rhZnumpyrZscipyrZpandasrZ matplotlibr‹Zmatplotlib.colorsrr-Zmatplotlib.pyplotZpyplotr9Z!external.six.moves.urllib.requestrrZexternal.six.moves.http_clientrZdistutils.versionr Ú __version__rvrŠÚ__all__rrr$r+r r r r;r rGrQrVrYr]r`rrprr€rr‰rr“r›ržrrrrÚsT       #  Y %    : '2