cMh ddlmZddlmZmZddlZddlmZmZddl Z ddl Z ddl Z ddl ZddlmZddlmZddlmZddlmcmZerddl mZ d.d/dZd0dZd1dZd2dZd3dZd4d Z d5d"Z!d6d$Z"d5d%Z#d7d&Z$d8d'Z%d8d(Z&d9d+Z'd:d-Z(dS);) annotations) CollectionIteratorN) TYPE_CHECKINGcast)MatplotlibColor)find_stack_level) is_list_like)Colormapdefault num_colorsintcolormapColormap | None color_typestrcolor3dict[str, Color] | Color | Collection[Color] | Nonect|tr|St||||}tt ||S)a Get standard colors based on `colormap`, `color_type` or `color` inputs. Parameters ---------- num_colors : int Minimum number of colors to be returned. Ignored if `color` is a dictionary. colormap : :py:class:`matplotlib.colors.Colormap`, optional Matplotlib colormap. When provided, the resulting colors will be derived from the colormap. color_type : {"default", "random"}, optional Type of colors to derive. Used if provided `color` and `colormap` are None. Ignored if either `color` or `colormap` are not None. color : dict or str or sequence, optional Color(s) to be used for deriving sequence of colors. Can be either be a dictionary, or a single color (single color string, or sequence of floats representing a single color), or a sequence of colors. Returns ------- dict or list Standard colors. Can either be a mapping if `color` was a dictionary, or a list of colors with a length of `num_colors` or more. Warns ----- UserWarning If both `colormap` and `color` are provided. Parameter `color` will override. rrrr r ) isinstancedict_derive_colorslist _cycle_colors)r rrrcolorss a/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/pandas/plotting/_matplotlib/style.pyget_standard_colorsrsYL%  F  f<<< = == Color | Collection[Color] | Nonestr | Colormap | Nonereturn list[Color]c||t||S|3|"tjdtt |St ||S)aa Derive colors from either `colormap`, `color_type` or `color` inputs. Get a list of colors either from `colormap`, or from `color`, or from `color_type` (if both `colormap` and `color` are None). Parameters ---------- color : str or sequence, optional Color(s) to be used for deriving sequence of colors. Can be either be a single color (single color string, or sequence of floats representing a single color), or a sequence of colors. colormap : :py:class:`matplotlib.colors.Colormap`, optional Matplotlib colormap. When provided, the resulting colors will be derived from the colormap. color_type : {"default", "random"}, optional Type of colors to derive. Used if provided `color` and `colormap` are None. Ignored if either `color` or `colormap`` are not None. num_colors : int Number of colors to be extracted. Returns ------- list List of colors extracted. Warns ----- UserWarning If both `colormap` and `color` are provided. Parameter `color` will override. NrzC'color' and 'colormap' cannot be used simultaneously. Using 'color') stacklevel)_get_colors_from_colormapwarningswarnr _get_colors_from_color_get_colors_from_color_typers rrrPsvN }-(jIIII     MU+--    &e,,,*:*MMMMr rIterator[Color]c#Kt|t|}tjtj||Ed{VdS)zCycle colors until achieving max of `num_colors` or length of `colors`. Extra colors will be ignored by matplotlib if there are more colors than needed and nothing needs to be done here. N)maxlen itertoolsislicecycle)rr max_colorss rrrsR ZV--J  7 7DDDDDDDDDDDr str | Colormapcht|fdtjdd|DS)zGet colors from colormap.c&g|] }|Sr7).0numcmaps r z-_get_colors_from_colormap..s! C C C#DDII C C Cr r)r9)_get_cmap_instancenplinspace)rr r:s @rr'r's= h ' 'D C C C CQz!B!B!B C C CCr r ct|tr)|}tj|}|t d|d|S)z$Get instance of matplotlib colormap.Nz Colormap z is not recognized)rrmpl colormaps ValueError)rr:s rr=r=sK(C  C=*  AAAABB B Or Color | Collection[Color]ct|dkrtd|t|rtt|}|gStt t|}t t|S)z!Get colors from user input color.rzInvalid color argument: )r/rC_is_single_colorrColorrr!_gen_list_of_colors_from_iterablers rr*r*s{ 5zzQ;E;;<<<UE""w E"E * *E 1%88 9 99r boolctt|trt|rdSt|rdSdS)a0Check if `color` is a single color, not a sequence of colors. Single color is of these kinds: - Named color "red", "C0", "firebrick" - Alias "g" - Sequence of floats, such as (0.1, 0.2, 0.3) or (0.1, 0.2, 0.3, 0.4). See Also -------- _is_single_string_color TF)rr_is_single_string_color_is_floats_colorrIs rrFrFsE%"9%"@"@tt 5r Collection[Color]c#^K|D]'}t|r|Vtd|dS)zS Yield colors from string of several letters or from collection of colors. zInvalid color N)rFrC)rxs rrHrHsS33 A   3GGGG1a1122 2 33r ctt|o>t|dkpt|dkotd|DS)zACheck if color comprises a sequence of floats representing color.c3NK|] }t|ttfV!dS)N)rrfloat)r8rPs r z#_is_floats_color..s0;; 1sEl++;;;;;;r )rJr r/allrIs rrMrMs_ U < ZZ1_ /E a < ;;U;;; ; ;  r ct|dkrt|S|dkrt|Std)z&Get colors from user input color type.r randomz/color_type must be either 'default' or 'random')_get_default_colors_get_random_colorsrC)rr s rr+r+sCY":... x  !*---JKKKr cPddlm}d|jdD}|d|S)z=Get `num_colors` of default colors from matplotlib rc params.rNcg|] }|d SrIr7)r8cs rr;z'_get_default_colors..s B B BQaj B B Br zaxes.prop_cycle)matplotlib.pyplotpyplotrcParams)r pltrs rrZrZs>###### B B#,/@"A B B BF !J, r c4dt|DS)z"Get `num_colors` of random colors.c,g|]}t|Sr7) _random_color)r8r9s rr;z&_get_random_colors..s < < <3M#   < < pandas._typingrrGpandas.util._exceptionsr pandas.core.dtypes.commonr pandas.core.commoncorecommonrjr rrrr'r=r*rFrHrMr+rZr[rerLr7r rrsS""""""333333444444222222 +****** !%AE 0>0>0>0>0>f1N1N1N1NhEEEEDDDD : : : :,3333LLLL    ==== r