U +if@sddlmZddlZddlmZmZmZmZmZm Z ddl Z ddl m Z ddl ZddlZddlmZddlmmZerddl mZeeeefZd;ddd d d d d Zddd dddddZddddddZddddddZdddddZd dd!d"d#Z d d$d!d%d&Z!d'dd!d(d)Z"d d$d!d*d+Z#d ddd,d-d.Z$ddd/d0d1Z%ddd/d2d3Z&dd4d5d6d7Z'd8d$d!d9d:Z(dS)<) annotationsN) TYPE_CHECKING CollectionIteratorSequenceUnioncast) is_list_like)ColormapdefaultintzColormap | Nonestrz3dict[str, Color] | Color | Collection[Color] | None) num_colorscolormap color_typecolorcCs.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. rrrrr) isinstancedict_derive_colorslist _cycle_colors)rrrrcolorsrT/home/mars/bis/venv/lib/python3.8/site-packages/pandas/plotting/_matplotlib/style.pyget_standard_colorss& rz Color | Collection[Color] | Nonezstr | Colormap | Nonez list[Color])rrrrreturncCsN|dkr|dk rt||dS|dk r>|dk r6tdt|St||dSdS)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')_get_colors_from_colormapwarningswarn_get_colors_from_color_get_colors_from_color_typerrrrrPs' rzIterator[Color])rrrccs*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)rrZ max_colorsrrrrsrzstr | Colormap)rrrcs&tfddtjdd|dDS)zGet colors from colormap.csg|] }|qSrr.0numrrr sz-_get_colors_from_colormap..r)r*)_get_cmap_instancenpZlinspace)rrrr+rrsrr )rrcCs4t|tr0|}t|}|dkr0td|d|S)z$Get instance of matplotlib colormap.Nz Colormap z is not recognized)rr cmZget_cmap ValueError)rZcmaprrrr.s   r.zColor | Collection[Color])rrcCsLt|dkrtd|t|r2tt|}|gSttt|}tt|S)z!Get colors from user input color.rzInvalid color argument: )r$r1_is_single_colorrColorrr!_gen_list_of_colors_from_iterablerrrrr!s  r!boolcCs&t|trt|rdSt|r"dSdS)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_colorr5rrrr2s r2zCollection[Color]ccs,|D]"}t|r|Vqtd|qdS)zS Yield colors from string of several letters or from collection of colors. zInvalid color N)r2r1)rxrrrr4sr4cCs6tt|o2t|dks"t|dko2tdd|DS)zACheck if color comprises a sequence of floats representing color.css|]}t|ttfVqdS)N)rr float)r)r9rrr sz#_is_floats_color..)r6r r$allr5rrrr8s r8)rrrcCs,|dkrt|S|dkr t|StddS)z&Get colors from user input color type.r randomz/color_type must be either 'default' or 'random'N)_get_default_colors_get_random_colorsr1)rrrrrr"s r")rrcCs,ddlm}dd|jdD}|d|S)z=Get `num_colors` of default colors from matplotlib rc params.rNcSsg|] }|dqSr5r)r)crrrr,sz'_get_default_colors..zaxes.prop_cycle)Zmatplotlib.pyplotZpyplotZrcParams)rZpltrrrrr@s r@cCsddt|DS)z"Get `num_colors` of random colors.cSsg|] }t|qSr) _random_colorr(rrrr,sz&_get_random_colors..)rangerrrrrAsrAz list[float])columnrcCst|}|dS)z4Get a random color represented as a list of length 3r:)comZ random_stateZrandtolist)rErsrrrrCs rCr3cCs8tj}z||Wntk r.YdSXdSdS)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. FTN) matplotlibrZColorConverterZto_rgbar1)rconvrrrr7s  r7)Nr N)) __future__rr%typingrrrrrrrZ matplotlib.cmr0Zmatplotlib.colorsrInumpyr/Zpandas.core.dtypes.commonr Zpandas.core.commoncorecommonrFr r r<r3rrrrr.r!r2r4r8r"r@rArCr7rrrrs8     33