o ۭj @s2ddlmZddlmZmZddlZddlmZmZddl Z ddl Z ddl Z ddl ZddlmZddlmZddlmZddlmmZerMddl mZ  dddZd?ddZd@dd ZdAd"d#ZdBd%d&Z dCd(d)Z!dDd+d,Z"dCd-d.Z#dEd/d0Z$dFd1d2Z%dFd3d4Z&dGd7d8Z'dHd:d;Z(dS)I) annotations) CollectionIteratorN) TYPE_CHECKINGcast)MatplotlibColor)find_stack_level) is_list_like)Colormapdefault num_colorsintcolormapColormap | None color_typestrcolor3dict[str, Color] | Color | Collection[Color] | NonecCs.t|tr|St||||d}tt||dS)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 rrrcolorsr^/var/www/html/CropPilot/venv/lib/python3.10/site-packages/pandas/plotting/_matplotlib/style.pyget_standard_colorss &r Color | Collection[Color] | Nonestr | Colormap | Nonereturn list[Color]cCsP|dur|durt||dS|dur"|durtjdtdt|St||dS)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_typerrrrrPs'  rrIterator[Color]ccs,t|t|}tt||EdHdS)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_colorsrrrrsrstr | Colormapcs&t|fddtjdd|dDS)zGet colors from colormap.csg|]}|qSrr.0numcmaprr z-_get_colors_from_colormap..r)r3)_get_cmap_instancenplinspace)rr rr4rr$sr$r cCs4t|tr|}tj|}|durtd|d|S)z$Get instance of matplotlib colormap.Nz Colormap z is not recognized)rrmpl colormaps ValueError)rr5rrrr9s  r9Color | Collection[Color]cCsLt|dkr td|t|rtt|}|gSttt|}tt|S)z!Get colors from user input color.rzInvalid color argument: )r+r>_is_single_colorrColorrr!_gen_list_of_colors_from_iterablerrrrr's   r'boolcCs&t|tr t|r dSt|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_colorrCrrrr@s  r@Collection[Color]ccs,|D]}t|r |Vqtd|dS)zS Yield colors from string of several letters or from collection of colors. zInvalid color N)r@r>)rxrrrrBs rBcCs6tt|ot|dkpt|dkotdd|DS)zACheck if color comprises a sequence of floats representing color.css|] }t|ttfVqdS)N)rr float)r2rHrrr sz#_is_floats_color..)rDr r+allrCrrrrFs rFcCs(|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_colorsr>)rr rrrr(s r(cCs,ddlm}dd|jdD}|d|S)z=Get `num_colors` of default colors from matplotlib rc params.rNcSsg|]}|dqSrCr)r2crrrr6r7z'_get_default_colors..zaxes.prop_cycle)matplotlib.pyplotpyplotrcParams)r pltrrrrrOs  rOcCsddt|DS)z"Get `num_colors` of random colors.cSsg|]}t|qSr) _random_colorr1rrrr6r7z&_get_random_colors..)rangerrrrrPsrPcolumn list[float]cCst|}|dS)z4Get a random color represented as a list of length 3rI)com random_staterandtolist)rXrsrrrrVs rVrAcCs0tj}z||WdStyYdSw)aCheck if `color` is a single string color. Examples of single string colors: - 'r' - 'g' - 'red' - 'green' - 'C3' - 'firebrick' Parameters ---------- color : Color Color string or sequence of floats. Returns ------- bool True if `color` looks like a valid color. False otherwise. FT) matplotlibrColorConverterto_rgbar>)rconvrrrrEs   rE)Nr N)r r rrrrrr) rrrr rrr r r!r")rr"r r r!r))rr0r r r!r")rr0r!r )rr?r!r")rr?r!rD)rrGr!r))rrr r r!r")r r r!r")rXr r!rY)rrAr!rD)) __future__rcollections.abcrrr,typingrrr%r_r<matplotlib.colorsnumpyr:pandas._typingrrApandas.util._exceptionsrpandas.core.dtypes.commonr pandas.core.commoncorecommonrZr rrrr$r9r'r@rBrFr(rOrPrVrErrrrs>      3 4