U
|
¬ý°d{< ã @ s¾ d dl mZ d dlmZ d dlmZmZmZ d dlZd dl m
|
Z
|
d dl Z d dl Z d dlmZ d dlmZ d dlmZmZmZ erÌd d lmZ d d
|
lmZ d d lmZ d d l m Z d d l m Z d dl m Z m!Z! dddddZ"dddddZ#dddddZ$dHdd d d d Z%dId!d"d#d$d%d&d'Z&dJd!dddd#d*d+d,Z'd-dd.d/d0Z(dd#dd1d2d3Z)d4d!d!d!d!dddd5d6d7Z*d8d9d:d;d<Z+dKd8d=d>d?Z,dd@ddAdBZ-dCdDdEdFdGZ.dS )Lé )Ú annotations)Úceil)Ú TYPE_CHECKINGÚIterableÚSequenceN)Úticker)Úfind_stack_level)Ú is_list_like)Ú ABCDataFrameÚABCIndexÚ ABCSeries)ÚAxes)ÚAxis)ÚFigure)ÚLine2D)ÚTable)Ú DataFrameÚSeriesr Úbool)ÚfigÚreturnc C s t | dsdS | ¡ S )z+Whether fig has constrained_layout enabled.Úget_constrained_layoutF)Úhasattrr )r © r úXd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/plotting/_matplotlib/tools.pyÚdo_adjust_figure& s
|
r ÚNonec O s t | r| j|| dS )zCCall fig.subplots_adjust unless fig has constrained_layout enabled.N)r Zsubplots_adjust)r ÚargsÚkwargsr r r Úmaybe_adjust_figure- s r r )Úaxr c C s: | ¡ D ]}| d¡ | |¡ q| ¡ }t|dd d S )NÚrightgÉ?)Úbottom)Úget_xticklabelsZset_haZ set_rotationÚ
|
get_figurer )r ZrotÚlabelr r r r Úformat_date_labels3 s
|
|
r& zDataFrame | Seriesr )Údatar c K sf t |tr| ¡ }nt |tr ntd|d kr6|j}|d krD|j}|j}tj j | f|||d|S )Nz&Input data must be DataFrame or Series)ÚcellTextÚ rowLabelsÚ colLabels)
|
Ú
|
isinstancer Zto_framer
|
Ú
|
ValueErrorÚindexÚcolumnsÚvaluesÚ
|
matplotlibÚtable)r r' r) r* r r( r r r r1 < s&
|
|
|
ÿ ÿÿr1 ÚboxÚintztuple[int, int] | NoneÚstrztuple[int, int])ÚnplotsÚlayoutÚ layout_typer c C sx |d k rÌt |ttfr"t|dkr*td|\}}|dkr\|dkr\t| | |f }\}}nF|dkr|dkr|t| | f }\}}n |dkr¢|dkr¢d}t||| | k rÈtd| d| d| |S |d krØd
|
S |d krèd | fS |d krø| d fS d
|
dddd}z
|
|| W S tk
|
rr d }|d | k r@|d 7 }q&|d | | krb||d f Y S ||f Y S Y nX d S )Né z)Layout must be a tuple of (rows, columns)éÿÿÿÿr z1At least one dimension of layout must be positivez
|
Layout of Úxz# must be larger than required size Zsingle)é r; Z
|
horizontalr; Zvertical)r; r8 )r8 r8 )r; r8 é é )r+ ÚtupleÚlistÚlenr, r ÚKeyError)r5 r6 r7 ÚnrowsÚncolsÚmsgZlayoutsÚkr r r Ú _get_layoutS s@ ÿ
|
rF FT)ÚnaxesÚsharexÚshareyÚsqueezer7 c K s ddl m} |dkri }|dkr.| jf |}
|
n¼t|r¦|rBt|}|dk r\tjdtt d |sd|rvtjdtt d |j | kr|j
|
d ¡ }
|
|
|
|fS t d| d| ¡ }
|
| dkrÐ|rÂ|
|
|fS |
|
t|fS ntjd tt d |
|
¡ t| ||d
|
\} } | | } tj| td }|
|
j| | df|}|r2||d < |r@||d < ||d< td| D ]F}| ¡ }|| krxd|d < d|d < |
|
j| | |d f|}|||< qR| | krÂ|| d
D ]}| d¡ q°t|| | | | || |r| dkrð|d }n| | | ¡ ¡ }n | | | ¡}|
|
|fS )a
|
Create a figure with a set of subplots already made.
|
|
This utility wrapper makes it convenient to create common layouts of
|
subplots, including the enclosing figure object, in a single call.
|
|
Parameters
|
----------
|
naxes : int
|
Number of required axes. Exceeded axes are set invisible. Default is
|
nrows * ncols.
|
|
sharex : bool
|
If True, the X axis will be shared amongst all subplots.
|
|
sharey : bool
|
If True, the Y axis will be shared amongst all subplots.
|
|
squeeze : bool
|
|
If True, extra dimensions are squeezed out from the returned axis object:
|
- if only one subplot is constructed (nrows=ncols=1), the resulting
|
single Axis object is returned as a scalar.
|
- for Nx1 or 1xN subplots, the returned object is a 1-d numpy object
|
array of Axis objects are returned as numpy 1-d arrays.
|
- for NxM subplots with N>1 and M>1 are returned as a 2d array.
|
|
If False, no squeezing is done: the returned axis object is always
|
a 2-d array containing Axis instances, even if it ends up being 1x1.
|
|
subplot_kw : dict
|
Dict with keywords passed to the add_subplot() call used to create each
|
subplots.
|
|
ax : Matplotlib axis object, optional
|
|
layout : tuple
|
Number of rows and columns of the subplot grid.
|
If not specified, calculated from naxes and layout_type
|
|
layout_type : {'box', 'horizontal', 'vertical'}, default 'box'
|
Specify how to layout the subplot grid.
|
|
fig_kw : Other keyword arguments to be passed to the figure() call.
|
Note that all keywords not recognized above will be
|
automatically included here.
|
|
Returns
|
-------
|
fig, ax : tuple
|
- fig is the Matplotlib Figure object
|
- ax can be either a single axis object or an array of axis objects if
|
more than one subplot was created. The dimensions of the resulting array
|
can be controlled with the squeeze keyword, see above.
|
|
Examples
|
--------
|
x = np.linspace(0, 2*np.pi, 400)
|
y = np.sin(x**2)
|
|
# Just a figure and one subplot
|
f, ax = plt.subplots()
|
ax.plot(x, y)
|
ax.set_title('Simple plot')
|
|
# Two subplots, unpack the output array immediately
|
f, (ax1, ax2) = plt.subplots(1, 2, sharey=True)
|
ax1.plot(x, y)
|
ax1.set_title('Sharing Y axis')
|
ax2.scatter(x, y)
|
|
# Four polar axes
|
plt.subplots(2, 2, subplot_kw=dict(polar=True))
|
r Nz6When passing multiple axes, layout keyword is ignored.)Ú
|
stacklevelzoWhen passing multiple axes, sharex and sharey are ignored. These settings must be specified when creating axes.z"The number of passed axes must be z , the same as the output plotr; zTTo output multiple subplots, the figure containing the passed axes is being cleared.)r6 r7 ©ZdtyperH rI F)Úmatplotlib.pyplotÚpyplotÚfigurer Ú flatten_axesÚwarningsÚwarnÚ UserWarningr ÚsizeZflatr$ r, ÚclearrF ÚnpÚemptyÚobjectZ add_subplotÚrangeÚcopyÚ set_visibleÚhandle_shared_axesZreshaperJ )rG rH rI rJ Z
|
subplot_kwr r6 r7 Zfig_kwÚpltr rB rC r5 ÚaxarrZax0ÚiÚkwdsÚaxesr r r Úcreate_subplots
s~ U ýü
|
|
ÿü
|
|
|
|
rb r )Úaxisr c C s | ¡ D ]}| d¡ qt| ¡ tjr6| t ¡ ¡ t| ¡ tj rV|
|
t d¡¡ | ¡ D ]}| d¡ q^| ¡ d¡ d S )NFÚ )Zget_majorticklabelsr[ r+ Zget_minor_locatorr Z NullLocatorZset_minor_locatorZ AutoLocatorZget_minor_formatterZ NullFormatterZset_minor_formatterZFormatStrFormatterZget_minorticklabelsZ get_label)rc Útr r r Ú_remove_labels_from_axis= s rf )Úax1Ú compare_axisr c C sj |dkr| ¡ }n|dkr$| ¡ }ntd| | ¡}| ¡ ¡ }|D ] }t || ¡ ¡ ¡sF dS qFdS )aA
|
Return whether an axis is externally shared.
|
|
Parameters
|
----------
|
ax1 : matplotlib.axes.Axes
|
Axis to query.
|
compare_axis : str
|
`"x"` or `"y"` according to whether the X-axis or Y-axis is being
|
compared.
|
|
Returns
|
-------
|
bool
|
`True` if the axis is externally shared. Otherwise `False`.
|
|
Notes
|
-----
|
If two axes with different positions are sharing an axis, they can be
|
referred to as *externally* sharing the common axis.
|
|
If two axes sharing an axis also have the same position, they can be
|
referred to as *internally* sharing the common axis (a.k.a twinning).
|
|
_handle_shared_axes() is only interested in axes externally sharing an
|
axis, regardless of whether either of the axes is also internally sharing
|
with a third axis.
|
r: ÚyzD_has_externally_shared_axis() needs 'x' or 'y' as a second parameterTF)Zget_shared_x_axesZget_shared_y_axesr, Z get_siblingsÚ get_positionZ
|
get_pointsrV Z array_equal)rg rh ra Z
|
ax1_pointsZax2r r r Ú_has_externally_shared_axisM s
|
|
ÿ
|
rk zIterable[Axes])r^ r5 rG rB rC rH rI r c C s6 |dkr2dd }dd }dd } |dkròz~t j|d |d ft jd}
|
| D ] } | ¡ |
|
|| || f< qL| D ]6} |
|
|| d || f sqn|st| drnt| j qnW nH tk
|
rð dd } | D ]&} | | rÒqÄ|sàt| drÄt| j qÄY nX |dkr2| D ]0} | | rq |s$t| d r t| j q d S )
|
Nr; c S s | ¡ jjS ©N)Úget_subplotspecZrowspanÚstart©r: r r r Ú<lambda> ó z$handle_shared_axes.<locals>.<lambda>c S s | ¡ jjS rl )rm Zcolspanrn ro r r r rp rq c S s | ¡ ¡ S rl )rm Ú is_first_colro r r r rp rq rL r: c S s | ¡ ¡ S rl )rm Ú is_last_rowro r r r rp ¢ rq ri ) rV ÚzerosZbool_Z get_visiblerk rf ZxaxisÚ
|
IndexErrorZyaxis) r^ r5 rG rB rC rH rI Zrow_numZcol_numrr r6 r rs r r r r\ s6
|
|
|
r\ zAxes | Sequence[Axes]z
|
np.ndarray)ra r c C s< t | st | g¡S t| tjtfr2t | ¡ ¡ S t | ¡S rl )r rV Úarrayr+ Zndarrayr ZasarrayZravel©ra r r r rP ´ s
|
rP rw c C s dd l m} t| D ]l}|d k r2|j| ¡ |d |d k rL|j| ¡ |d |d k rf|j| ¡ |d |d k r|j| ¡ |d q| S )Nr )Zfontsize)Zrotation)rM rN rP Zsetpr# Zget_yticklabels)ra Z
|
xlabelsizeZxrotZ
|
ylabelsizeZyrotr] r r r r Úset_ticks_props¼ s rx z list[Line2D]c C s< | ¡ }t| dr || j ¡ 7 }t| dr8|| j ¡ 7 }|S )NÚright_axÚleft_ax)Ú get_linesr ry rz )r Úlinesr r r Ú get_all_linesÑ s
|
|
r} zIterable[Line2D]ztuple[float, float])r| r c C sN t jt j }}| D ]0}|jdd}tt |¡|}tt |¡|}q||fS )NF)Úorig)rV ÚinfZ get_xdataÚminZnanminÚmaxZnanmax)r| Úleftr! Úliner: r r r Úget_xlimÝ s r )NN)Nr2 )FFTNNNr2 )NNNN)/Ú
|
__future__r Úmathr Útypingr r r rQ r0 r Zmatplotlib.tableÚnumpyrV Zpandas.util._exceptionsr Zpandas.core.dtypes.commonr Zpandas.core.dtypes.genericr
|
r r Zmatplotlib.axesr Zmatplotlib.axisr Zmatplotlib.figurer Zmatplotlib.linesr r Zpandasr r r r r& r1 rF rb rf rk r\ rP rx r} r r r r r Ú<module> sZ
|
ÿ ý4 ø 92 5
|
û
|
