a Sic8@sddlZddlmZddlmZzddlmZm Z m Z Wne yRddZYn0ddl m Z ddlmZmZmZmZmZgd Zd d Zd d ZddZdddZdddZdddZdddZd ddZdS)!N)LinearSegmentedColormap)interact FloatSlider IntSlidercCsd}t|dS)NzBInteractive palettes require `ipywidgets`, which is not installed.) ImportError)fmsgr K/var/www/html/django/DPS/env/lib/python3.9/site-packages/seaborn/widgets.pyrsr)palplot) color_palette dark_palette light_palettediverging_palettecubehelix_palette)choose_colorbrewer_palettechoose_cubehelix_palettechoose_dark_palettechoose_light_palettechoose_diverging_palettecCs*tdd}td|}|||S)zACreate a matplotlib colormap that will be updated by the widgets.Greys interactive)r r from_list_init _set_extremes)Zgreyscmapr r r _init_mutable_colormaps   rcCs||jdd<|dS)z8Change the LUT values in a matplotlib colormap in-place.Nr)_lutr)rcolorsr r r _update_lutsr!cCsddlm}|d tjdd\}}Wdn1s:0Y|jggdtddd tjddf}|j||d dS) z&Show a continuous matplotlib colormap.r ) axes_stylewhite)g @g?)figsizeN)xticksyticksrr)r) rcmodr"pltsubplotssetnplinspacenewaxis pcolormesh)rr"raxxr r r _show_cmap%s   .r1Fcs|drrtdgr$t|drhgd}gd}t|dtdddd |ffd d }nx|d rgd }ddg}t|dtdddd |ffdd }n4|drgd}t|dtdddd ffdd }rSS)avSelect a palette from the ColorBrewer set. These palettes are built into matplotlib and can be used by name in many seaborn functions, or by passing the object returned by this function. Parameters ---------- data_type : {'sequential', 'diverging', 'qualitative'} This describes the kind of data you want to visualize. See the seaborn color palette docs for more information about how to choose this value. Note that you can pass substrings (e.g. 'q' for 'qualitative. as_cmap : bool If True, the return value is a matplotlib colormap rather than a list of discrete colors. Returns ------- pal or cmap : list of colors or matplotlib colormap Object that can be passed to plotting functions. See Also -------- dark_palette : Create a sequential palette with dark low values. light_palette : Create a sequential palette with bright low values. diverging_palette : Create a diverging palette from selected colors. cubehelix_palette : Create a sequential palette or colormap using the cubehelix system. qz)Qualitative palettes cannot be colormaps.s)rRedsGreensBluesOrangesPurplesBuGnBuPuGnBuOrRdPuBuPuRdRdPuYlGnPuBuGnYlGnBuYlOrBrYlOrRd)regularreversedark)rr minmaxvaluecsv|dkr|d7}n|dkr"|d7}rVt|d|}ttj|tdftnt|||dd<tdS)NrF_rrG_drr r!r+c_onesr1r namendesatvariantr as_cmaprpalr r choose_sequential\s   z5choose_colorbrewer_palette..choose_sequentiald)RdBuRdGyPRGnPiYGBrBGRdYlBuRdYlGnSpectralrErFrHcsd|dkr|d7}rDt|d|}ttj|tdftnt|||dd<tdS)NrFrNrrPrSrXr r choose_divergingrs  z4choose_colorbrewer_palette..choose_diverging)Set1Set2Set3PairedAccentPastel1Pastel2Dark2cs t|||dd<tdS)N)r r )rTrUrV)rZr r choose_qualitativesz6choose_colorbrewer_palette..choose_qualitative) startswith ValueErrorrrr) data_typerYoptsvariantsr[rgrpr rXr r/s<       rhuslcsxgrt|dkr.tdfdd }n>|dkrNtdfdd }n|d krltdfd d }rtSS)a4Launch an interactive widget to create a dark sequential palette. This corresponds with the :func:`dark_palette` function. This kind of palette is good for data that range between relatively uninteresting low values and interesting high values. Requires IPython 2+ and must be used in the notebook. Parameters ---------- input : {'husl', 'hls', 'rgb'} Color space for defining the seed value. Note that the default is different than the default input for :func:`dark_palette`. as_cmap : bool If True, the return value is a matplotlib colormap rather than a list of discrete colors. Returns ------- pal or cmap : list of colors or matplotlib colormap Object that can be passed to plotting functions. See Also -------- dark_palette : Create a sequential palette with dark low values. light_palette : Create a sequential palette with bright low values. cubehelix_palette : Create a sequential palette or colormap using the cubehelix system. rgbgg?csR|||f}r0t|ddd}t|tnt||dddd<tdSNrrwinputrr!r1r rgbrUcolorr rXr r choose_dark_palette_rgbs   z4choose_dark_palette..choose_dark_palette_rgbhlscsR|||f}r0t|ddd}t|tnt||dddd<tdSNrrr}rhlr3rUrr rXr r choose_dark_palette_hlss   z4choose_dark_palette..choose_dark_palette_hlsrvrgrccsR|||f}r0t|ddd}t|tnt||dddd<tdSNrrvr}rrr3rrUrr rXr r choose_dark_palette_husls   z5choose_dark_palette..choose_dark_palette_husl)rxrxrxry)rxrxrxry)rrrryrr)r~rYrrrr rXr rs6   rcsxgrt|dkr.tdfdd }n>|dkrNtdfdd }n|d krltdfd d }rtSS)a7Launch an interactive widget to create a light sequential palette. This corresponds with the :func:`light_palette` function. This kind of palette is good for data that range between relatively uninteresting low values and interesting high values. Requires IPython 2+ and must be used in the notebook. Parameters ---------- input : {'husl', 'hls', 'rgb'} Color space for defining the seed value. Note that the default is different than the default input for :func:`light_palette`. as_cmap : bool If True, the return value is a matplotlib colormap rather than a list of discrete colors. Returns ------- pal or cmap : list of colors or matplotlib colormap Object that can be passed to plotting functions. See Also -------- light_palette : Create a sequential palette with bright low values. dark_palette : Create a sequential palette with dark low values. cubehelix_palette : Create a sequential palette or colormap using the cubehelix system. rwrxrycsR|||f}r0t|ddd}t|tnt||dddd<tdSr|rr!r1r rrXr r choose_light_palette_rgbs   z6choose_light_palette..choose_light_palette_rgbrcsR|||f}r0t|ddd}t|tnt||dddd<tdSrrrrXr r choose_light_palette_hlss   z6choose_light_palette..choose_light_palette_hlsrvrrcsR|||f}r0t|ddd}t|tnt||dddd<tdSrrrrXr r choose_light_palette_husl&s   z7choose_light_palette..choose_light_palette_husl)rxrxrxry)rxrxrxry)rrrryr)r~rYrrrr rXr rs6   rc stgrtttddddtddddtddddtddddtd dddd d d gffd d }rpSS)aLaunch an interactive widget to choose a diverging color palette. This corresponds with the :func:`diverging_palette` function. This kind of palette is good for data that range between interesting low values and interesting high values with a meaningful midpoint. (For example, change scores relative to some baseline value). Requires IPython 2+ and must be used in the notebook. Parameters ---------- as_cmap : bool If True, the return value is a matplotlib colormap rather than a list of discrete colors. Returns ------- pal or cmap : list of colors or matplotlib colormap Object that can be passed to plotting functions. See Also -------- diverging_palette : Create a diverging color palette or colormap. choose_colorbrewer_palette : Interactively choose palettes from the colorbrewer set, including diverging palettes. rrrJ rJ2r relightrGcsTr,t|||||d|}t|tn$t|||||||dd<tdSNr)rr!r1r )h_negh_posr3rseprUcenterr rXr r rYs   z:choose_diverging_palette..choose_diverging_palette)rrr)rYrr rXr r9s*   rc sgrtttddddtddddtddd dtdd ddtddd dtddd dtddd ddffdd }rSS)aLaunch an interactive widget to create a sequential cubehelix palette. This corresponds with the :func:`cubehelix_palette` function. This kind of palette is good for data that range between relatively uninteresting low values and interesting high values. The cubehelix system allows the palette to have more hue variance across the range, which can be helpful for distinguishing a wider range of values. Requires IPython 2+ and must be used in the notebook. Parameters ---------- as_cmap : bool If True, the return value is a matplotlib colormap rather than a list of discrete colors. Returns ------- pal or cmap : list of colors or matplotlib colormap Object that can be passed to plotting functions. See Also -------- cubehelix_palette : Create a sequential palette or colormap using the cubehelix system. rHrf rJrrzr g?g?g333333?g333333?Fc shr>td|||||||}ttj|tdftn&t||||||||dd<tdSr)rr!r+rQrRr1r ) n_colorsstartrotgammahuerrGrFr rXr r choose_cubehelixs     z2choose_cubehelix_palette..choose_cubehelix)rrrr)rYrr rXr rts        r)F)rvF)rvF)F)F)numpyr+matplotlib.pyplotpyplotr(matplotlib.colorsrZ ipywidgetsrrrrmiscplotr palettesr rrrr__all__rr!r1rrrrrr r r r s"      ` U U ;