U
|
¬ý°dæ ã @ s d dl mZ d dlZd dlmZmZmZmZ d dlZd dl Z
|
d dl Z d dl Z d dlmZ d dlmZ d dlmZ d dlm mZ erd dl mZ d=d d
|
d d d ddZddd d ddddZdd ddddZ dd ddd d Z dd d d d!Z d"dd#d$d%Z d"d&d#d'd(Z d)dd#d*d+Z!d"d&d#d,d-Z"d d dd.d/d0Z#d dd1d2d3Z$d dd1d4d5Z%d d6d7d8d9Z&d:d&d#d;d<Z'dS )>é )Ú annotationsN)Ú TYPE_CHECKINGÚ
|
CollectionÚIteratorÚcast)ÚMatplotlibColor)Úfind_stack_level)Ú is_list_like)ÚColormapÚdefaultÚintzColormap | NoneÚstrz3dict[str, Color] | Color | Collection[Color] | None)Ú
|
num_colorsÚcolormapÚ
|
color_typeÚcolorc C s. t |tr|S t|||| 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.
|
©r r r r ©r )Ú
|
isinstanceÚdictÚ_derive_colorsÚlistÚ _cycle_colors)r r r r Úcolors© r úXd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/plotting/_matplotlib/style.pyÚget_standard_colors s &
|
ür z Color | Collection[Color] | Nonezstr | Colormap | Nonez list[Color])r r r r Úreturnc C sT | dkr |dk r t ||dS | dk rD|dk r<tjdt d t| S t||dS 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.
|
Nr zC'color' and 'colormap' cannot be used simultaneously. Using 'color')Ú
|
stacklevel)Ú_get_colors_from_colormapÚwarningsÚwarnr Ú_get_colors_from_colorÚ_get_colors_from_color_typer r r r r N s ' þr zIterator[Color])r r r c c s* t |t| }t t | ¡|¡E dH dS )zÈCycle 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)ÚmaxÚlenÚ itertoolsÚisliceÚcycle)r r Z
|
max_colorsr r r r s r zstr | Colormap)r r r c s& t | fddtjdd|dD S )zGet colors from colormap.c s g | ] } |qS r r ©Ú.0Únum©Úcmapr r Ú
|
<listcomp> s z-_get_colors_from_colormap.<locals>.<listcomp>r é )r+ )Ú_get_cmap_instanceÚnpZlinspace)r r r r, r r s r r
|
)r r c C s4 t | tr0| }tj| } | dkr0td| d| S )z$Get instance of matplotlib colormap.Nz Colormap z is not recognized)r r ÚmplZ colormapsÚ
|
ValueError)r r- r r r r0 s
|
|
r0 zColor | Collection[Color])r r c C sL t | dkrtd| t| r2tt| } | gS ttt | } tt| S )z!Get colors from user input color.r zInvalid color argument: )r% r3 Ú_is_single_colorr ÚColorr r Ú!_gen_list_of_colors_from_iterable©r r r r r" s
|
r" Úboolc C s& t | trt| rdS t| r"dS dS )a0 Check 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)r r Ú_is_single_string_colorÚ_is_floats_colorr7 r r r r4 ® s
|
r4 zCollection[Color]c c s, | D ]"}t |r|V qtd| qdS )zS
|
Yield colors from string of several letters or from collection of colors.
|
zInvalid color N)r4 r3 )r Úxr r r r6 Ä s r6 c C s6 t t| o2t| dks"t| dko2tdd | D S )zACheck if color comprises a sequence of floats representing color.é é c s s | ]}t |ttfV qd S )N)r r Úfloat)r* r; r r r Ú <genexpr>Ô s z#_is_floats_color.<locals>.<genexpr>)r8 r r% Úallr7 r r r r: Ï s ÿýr: )r r r c C s, | dkrt |S | dkr t|S tddS )z&Get colors from user input color type.r Úrandomz/color_type must be either 'default' or 'random'N)Ú_get_default_colorsÚ_get_random_colorsr3 )r r r r r r# Ø s
|
r# )r r c C s, ddl m} dd |jd D }|d|
S )z=Get `num_colors` of default colors from matplotlib rc params.r Nc S s g | ] }|d qS r7 r )r* Úcr r r r. æ s z'_get_default_colors.<locals>.<listcomp>zaxes.prop_cycle)Zmatplotlib.pyplotZpyplotZrcParams)r Zpltr r r r rB â s rB c C s dd t | D S )z"Get `num_colors` of random colors.c S s g | ] }t |qS r )Ú _random_colorr) r r r r. ì s z&_get_random_colors.<locals>.<listcomp>)Úranger r r r rC ê s rC z list[float])Úcolumnr c C s t | ¡}| d¡ ¡ S )z4Get a random color represented as a list of length 3r< )ÚcomZ random_stateZrandÚtolist)rG Úrsr r r rE ï s
|
rE r5 c C s8 t j ¡ }z| | ¡ W n tk
|
r. Y dS X dS dS )a Check 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)Ú
|
matplotlibr ZColorConverterZto_rgbar3 )r Úconvr r r r9 ö s
|
r9 )Nr N)(Ú
|
__future__r r& Útypingr r r r r rK r2 Zmatplotlib.colorsÚnumpyr1 Zpandas._typingr r5 Zpandas.util._exceptionsr Zpandas.core.dtypes.commonr Zpandas.core.commonÚcoreÚcommonrH r
|
r r r r r0 r" r4 r6 r: r# rB rC rE r9 r r r r Ú<module> s: ü34
|
|
|
|
