a 鴫j @sddlmZddlmZmZddlZddlmZmZddl Z ddl Z ddl Z ddl ZddlmZddlmZddlmZddlmmZerddl mZd>d d d d dddZddd d ddddZdd ddddZdd ddddZddd d!d"Zd#dd$d%d&Z d#d'd$d(d)Z!d*dd$d+d,Z"d#d'd$d-d.Z#d d dd/d0d1Z$d dd2d3d4Z%d dd2d5d6Z&d d7d8d9d:Z'd;d'd$dz-_get_colors_from_colormap..r)r+)_get_cmap_instancenpZlinspace)rrrr,rrsrr )rrcCs4t|tr0|}tj|}|dur0td|d|S)z$Get instance of matplotlib colormap.Nz Colormap z is not recognized)rr mplZ colormaps ValueError)rr-rrrr1s   r1zColor | Collection[Color])rrcCsLt|dkrtd|t|r2tt|}|gSttt|}tt|S)z!Get colors from user input color.rzInvalid color argument: )r%r4_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_colorr8rrrr5s r5zCollection[Color]ccs,|D]"}t|r|Vqtd|qdS)zS Yield colors from string of several letters or from collection of colors. zInvalid color N)r5r4)rxrrrr7sr7cCs6tt|o2t|dks"t|dko2tdd|DS)zACheck if color comprises a sequence of floats representing color.css|]}t|ttfVqdS)N)rr float)r*r<rrr r/z#_is_floats_color..)r9r r%allr8rrrr;s r;)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_colorsr4)rrrrrr#s r#)rrcCs,ddlm}dd|jdD}|d|S)z=Get `num_colors` of default colors from matplotlib rc params.rNcSsg|] }|dqSr8r)r*crrrr.r/z'_get_default_colors..zaxes.prop_cycle)Zmatplotlib.pyplotZpyplotZrcParams)rZpltrrrrrCs rCcCsddt|DS)z"Get `num_colors` of random colors.cSsg|] }t|qSr) _random_colorr)rrrr.r/z&_get_random_colors..)rangerrrrrDsrDz list[float])columnrcCst|}|dS)z4Get a random color represented as a list of length 3r=)comZ random_stateZrandtolist)rHrsrrrrFs rFr6cCs6tj}z||Wnty,YdS0dSdS)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_rgbar4)rconvrrrr:s   r:)Nr N)) __future__rcollections.abcrrr&typingrrr rLr3Zmatplotlib.colorsnumpyr2Zpandas._typingrr6Zpandas.util._exceptionsrZpandas.core.dtypes.commonr Zpandas.core.commoncorecommonrIr rrrrr1r"r5r7r;r#rCrDrFr:rrrrs<     34