o g/c@sdZddlZddlZddlZddlZddlZddlmZmZddl Z ddl m Z ddl ZddlZddlmZddlmZddlmZgdZddZd d ZdEd d ZddZddZddZdFddZ ddZ!ddZ"ddZ#  dGddZ$d d!Z%d"d#Z&dHd$d%Z'dId'd(Z(d)d*Z)d+d,Z*d-d.Z+dHd/d0Z,dJd1d2Z-d3d4Z.d5d6Z/d7d8Z0d9d:Z1d;d<Z2d=d>Z3d?d@Z4dAdBZ5dCdDZ6dS)Kz+Utility functions, mostly for internal use.N)urlopen urlretrieve)stats)normalize_kwargs) desaturatesaturateset_hls_values move_legenddespineget_dataset_names get_data_home load_datasetcOsDd}t|z |j|i|WSty!|j|i|YSw)z=Wrapper to handle different pandas sorting API pre/post 0.17.CThis function is deprecated and will be removed in a future version)warningswarn sort_valuesAttributeErrorsort)dfargskwargsmsgrK/home/ubuntu/cloudmapper/venv/lib/python3.10/site-packages/seaborn/utils.pysort_dfs  rc Csrt|dd}t|}g}tt|D]\}\}}||}||}||}|||gqt|j}|S)aConvert 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 ) np atleast_2dreshape atleast_1d enumerate transposeappendasarrayT) cisheightserrsizeilowhighhelowehighrrr ci_to_errsize s  r/ cCsNd}t|tt||\}}||}|d|d}|dd||fS)aReturn arguments to plt.bar for pmf-like histogram of an array. DEPRECATED: will be removed in a future version. 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)rr FutureWarningr histogramsum)abinsrnxr,wrrrpmf_hist>s   r:cCs@|j|jrz ||jWdStyYdSwdS)z>Force draw of a matplotlib figure, accounting for back-compat.N)canvasdrawstale get_rendererr)figrrr _draw_figure\s  r@cCsZd|kr dkstdtdtj|}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 rr1zprop must be between 0 and 1) ValueErrormplcolcolorConverterto_rgbcolorsys rgb_to_hls hls_to_rgb)colorproprgbr,ls new_colorrrrrgs rcCs 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 r1)rL)r)rHrrrrs rcCsPtj|}ttj|}t|||gD] \}}|dur |||<qtj|}|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)rBrCrDlistrErFr!rG)rHr,rKrLrJvalsr)valrrrrs  rcKs@d}t|tt}|j|fi||j|fi|dS)z[Grab current axis and label it. DEPRECATED: will be removed in a future version. rN)rrr2pltgca set_xlabel set_ylabel)xlabelylabelrraxrrraxlabels  rXcCs|t|S)aGHelper method for removing null values from data vectors. Parameters ---------- vector : vector object Must implement boolean masking with [] subscript syntax. Returns ------- clean_clean : same type as ``vector`` Vector of data with null values removed. May be a copy or a view. )pdnotnull)vectorrrr remove_nasr\cCs&tjd}d|jvr|dSdgS)a Return the list of colors in the current matplotlib color cycle Parameters ---------- None Returns ------- colors : list List of matplotlib colors in the current cycle, or dark gray if the current color cycle is empty. zaxes.prop_cyclerHz.15)mplrcParamskeysby_key)cyclerrrrget_color_cycles rbTFc Cs|dur|durtj}n|dur|j}n|dur|g}|D]H} dD]4} t|  } | j| | |durX| rXz|| d} Wn tyM|} Ynw| j| d| fq$|r|st dd| j j D} t dd| j j D}| j d| j j D]}|j| q}| j j D]}|j|q|r|st d d| jj D} t d d| jj D}| j d | jj D]}|j| q| jj D]}|j|q|rht| }|jrt|t| k|d}t|t| k|d }| jd ||| jd |||||k}|||k}| |t| }|jrht|t| k|d}t|t| k|d }| jd||| jd|||||k}|||k}| |qdS)a Remove the top and right spines from plot(s). fig : matplotlib figure, optional Figure to despine all axes of, defaults to the current figure. ax : matplotlib axes, optional Specific axes object to despine. Ignored if fig is provided. 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)toprightleftbottomroutwardcs|]}|jVqdSN tick1line get_visible.0trrr   zdespine..csrhrirjrmrrrrprqrdcsrhrirjrmrrrrp"rqcsrhrirjrmrrrrp&rqrcrrfre)rQgcfaxeslocalsspines set_visiblegetr set_positionanyyaxis majorTicks minorTicksset_ticks_position tick2linexaxisrr$ get_xtickssizecompressminget_xlimmax set_bounds set_xticks get_yticksget_ylim set_yticks)r?rWrcrdrerfoffsettrimrsax_iside is_visiblerPmaj_onmin_onroxticks firstticklastticknewticksyticksrrrr s             r c sddlm}t||r|j}|jj}n*t|tjjr!|j}|j}nt|tjj r7|j r1|j d}nd}|j}nd}t ||durJ|d}t ||j }dd|D}ttjjjfd d |D} | d | d } d |vr| |d d d |D} | D]\} } | jdi| dd| i|| q|d|j| |||||fd|i| }|| | t||r||_ dSdS)a< Recreate a plot's legend at a new location. The name is a slight misnomer. Matplotlib legends do not expose public control over their position parameters. So this function creates a new legend, copying over the data from the original object, which is then removed. Parameters ---------- obj : the object with the plot This argument can be either a seaborn or matplotlib object: - :class:`seaborn.FacetGrid` or :class:`seaborn.PairGrid` - :class:`matplotlib.axes.Axes` or :class:`matplotlib.figure.Figure` loc : str or int Location argument, as in :meth:`matplotlib.axes.Axes.legend`. kwargs Other keyword arguments are passed to :meth:`matplotlib.axes.Axes.legend`. Examples -------- .. include:: ../docstrings/move_legend.rst r)GridrNzC`obj` must be a seaborn Grid or matplotlib Axes or Figure instance.z has no legend attached.cSg|]}|qSr)get_textrmrrr zmove_legend..csi|] \}}|vr||qSrrrnkv legend_kwsrr szmove_legend..bbox_to_anchortitlecSs i|] \}}|dr||qS)title_) startswithrrrrrs frameonlocr)!seaborn.axisgridr isinstancelegendfigurer]rsAxeslegend_Figurelegends TypeErrorrA legendHandles get_textsinspect signatureLegend parameters propertiesitemspopset_textset setdefault legendPatchrlupdateremove set_titlerget_fontproperties_legend)objrrr old_legend legend_funcerrhandleslabelspropsr title_kwargskeyrP new_legendrrrr KsJ          r cCsFt||||d}t||||d}t|||}|S)z0Establish support for a kernel density estimate.rr1)rrrlinspace)databwgridsizecutclip support_min support_maxsupportrrr _kde_supportsrc Csd}t|tg}zt|}Wnty|g}d}Ynwt|D]\}}|dur4t||}n t tj|||}| |q#t |}|sN| }|S)aLike scoreatpercentile but can take and return array of percentiles. DEPRECATED: will be removed in a future version. 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`` rrN)rrr2lenrr!rscoreatpercentileravelrapply_along_axisr#r$squeeze) r5pctsaxisrscoresr7r)pscorerrr percentiless$     r_cCs&d|dd|df}t|||S)z2Return a percentile range from an array of values.2r)r nanpercentile)r5whichrrrrrcisrcCsDd}t|t|dkrdS|dkrdS|dkrdS|dkr d Sd S) z{Return a R-style significance string corresponding to p values. DEPRECATED: will be removed in a future version. rgMbP?z***g{Gz?z**g?*g?.)rrr2)rrrrr sig_starss rcCs:d}t|tt|}t|d}t|d}||S)zfCalculate the IQR for an array of numbers. DEPRECATED: will be removed in a future version. rK)rrr2rr$rr)r5rq1q3rrriqrs     rcCsLd}t| }|}Wdn1swYd}t||}|S)zkReport available example datasets, useful for reporting issues. Requires an internet connection. z'https://github.com/mwaskom/seaborn-dataNz+/mwaskom/seaborn-data/blob/master/(\w*).csv)rreadrefindalldecode)urlresphtmlpatdatasetsrrrr s  r cCsF|durtjdtjdd}tj|}tj|s!t||S)aReturn a path to the cache directory for example datasets. This directory is then used by :func:`load_dataset`. If the ``data_home`` argument is not specified, it tries to read from the ``SEABORN_DATA`` environment variable and defaults to ``~/seaborn-data``. N SEABORN_DATA~z seaborn-data)osenvironrwpathjoin expanduserexistsmakedirs) data_homerrrr s     r c Ks<t|tjr d}t|d|d}|rhttps://raw.githubusercontent.com/mwaskom/seaborn-data/master/z.csv'z%' is not one of the example datasets.rNtipsday)ThurFriSatSunsexMaleFemaletimeLunchDinnersmokerYesNoflightsmonthexercise)z1 minz15 minz30 minkind)restwalkingrunningdietzno fatzlow fattitanicclass)FirstSecondThirddeckABCDEFGpenguinsdiamondsrH)DEFGHIJclarity)IFVVS1VVS2VS1VS2SI1SI2I1r)IdealPremiumz Very GoodGoodFair)rrY DataFramerrrrr basenamerr rArread_csvilocisnullall CategoricalstruniquerNr) namecacherkwsrr cache_path full_pathrmonthsrrrr "sX #        r csL|sdSzdd|DfddD}t|dkWSty%YdSw)zReturn a boolean for whether the list of ticklabels have overlaps. Parameters ---------- labels : list of matplotlib ticklabels Returns ------- overlap : boolean True if any of the labels overlap. FcSrr)get_window_extentrnrKrrrrrz+axis_ticklabels_overlap..csg|]}|qSr)count_overlaps)rnbbboxesrrrsr1)r RuntimeError)roverlapsrrMraxis_ticklabels_overlaps  rQcCst|t|fS)zReturn 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. )rQget_xticklabelsget_yticklabels)rWrrraxes_ticklabels_overlaps rTcs|j|}fdd|D}Gfddd}t|tjjr'tjntj|_ |fdd|D}||fS)z=Return levels and formatted levels for brief numeric legends.cs(g|]}|dkr|dkr|qS)rr1rrJlimitsrrrs(z-locator_to_legend_entries..cseZdZfddZdS)z-locator_to_legend_entries..dummy_axiscsSrir)selfrUrrget_view_intervalsz?locator_to_legend_entries..dummy_axis.get_view_intervalN)__name__ __module__ __qualname__rXrrUrr dummy_axissr\csg|]}|qSrr)rnr8) formatterrrrr) tick_valuesastyperr]ticker LogLocator LogFormatterScalarFormatterrset_locs)locatorrVdtype raw_levelsr\formatted_levelsr)r]rVrlocator_to_legend_entriess   ricCsptjj|ddddf}t|dk|d|ddd}|gd}z|WSty7|YSw) 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 Nrg#?gףp= )@g)\(?gzG?g333333@)gz6?g,C?g]m{?) r]colorsrC to_rgba_arrayrwheredotitemrA)rHrJlumrrrrelative_luminances "  rpcCs8t|tr|Sz|jddWStyt|YSw)aReturn a string representing a Python object. Strings (i.e. type ``str``) are returned unchanged. Byte strings (i.e. type ``bytes``) are returned as UTF-8-decoded strings. For other objects, the method ``__str__()`` is called, and the result is returned as a string. Parameters ---------- obj : object Any Python object Returns ------- s : str UTF-8-decoded string representation of ``obj`` zutf-8)encoding)rrArr)rrrrto_utf8s   rrc CsZdgdgdgdgdgdgdgdgd gd }zt||}W|Sty,t||}Y|Sw) z>Wrapper for mpl.cbook.normalize_kwargs that supports <= 3.2.1.clwlsfcecmfcmecmewms) rH linewidth linestyle facecolor edgecolormarkerfacecolormarkeredgecolormarkeredgewidth markersize)rr)rEartist _alias_maprrr_normalize_kwargss"  rcCs(||vrtd|d|d|ddS)z+Raise if value for param is not in options.`z` must be one of z, but z was passed.`N)rA)paramoptionsvaluerrr_check_arguments rcCsDt|j}t|j}|D]}||vr||vr||j||<q|S)zBAssign default kwargs for call_func using values from source_func.)rrrdefault)rE call_func source_funcneededdefaultsrrrr_assign_default_kwargss rcCstjdd}|tjjd}|D]*}|\}}|}tdd|Ds>| d|D] }|dur=| |q2qdS)z@Make invisible-handle "subtitles" entries look more like titles.zlegend.title_fontsizeNrcss|]}|VqdSri)rl)rnrrrrrp1sz*adjust_legend_subtitles..) rQr^rwfindobjr] offsetboxVPacker get_childrenr? set_widthset_size)r font_sizehpackershpack draw_area text_areartextrrradjust_legend_subtitles)s    r)r0)NNN)NNTTFFNFri)rN)TN)7__doc__rrrrrEurllib.requestrrnumpyrscipyrpandasrY matplotlibr]matplotlib.colorsrjrBmatplotlib.pyplotpyplotrQmatplotlib.cbookr__all__rr/r:r@rrrrXr\rbr r rrrrrr r r rQrTrirprrrrrrrrrrs\      #   fZ *  b