U
|
¬ý°dÞ ã @ s d Z ddlmZ ddlmZmZ ddlmZmZm Z ddl
|
Z
|
ddl Z ddl mZ ddlmZ ddlmZmZ dd lmZ ddlm mZ dd
|
lmZ er¾dd l m Z dd l m Z dd l m!Z! ddddZ"ddddddZ#ddddZ$ddd d Z%e%d d Z&d d! Z'd"d# Z(dS )$z!
|
Core eval alignment algorithms.
|
é )Ú annotations)ÚpartialÚwraps)Ú TYPE_CHECKINGÚCallableÚSequenceN)ÚPerformanceWarning)Úfind_stack_level)Ú ABCDataFrameÚ ABCSeries)Ú PandasObject)Úresult_type_many)ÚF)ÚNDFrame)ÚIndexz7tuple[partial | type[NDFrame], dict[str, Index] | None])Úreturnc C sR d }t | jtjr&ttj| jjd}n$t| j}t| jdrJt || jj
|
}||fS )N)ÚdtypeÚaxes) Ú
|
isinstanceÚvalueÚnpÚndarrayr Z
|
asanyarrayr ÚtypeÚhasattrÚ_zip_axes_from_typer )Útermr Útyp© r úTd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/core/computation/align.pyÚ_align_core_single_unary_op& s
|
r z type[NDFrame]zSequence[Index]zdict[str, Index])r Únew_axesr c s fddt | jD S )Nc s i | ]\}}| | qS r r )Ú.0ÚiÚname©r r r Ú
|
<dictcomp>9 s z'_zip_axes_from_type.<locals>.<dictcomp>)Ú enumerateZ _AXIS_ORDERS)r r r r$ r r 6 s r Úboolc C s t dd | D S )zB
|
Check a sequence of terms for instances of PandasObject.
|
c s s | ]}t |jtV qd S ©N)r r r ©r! r r r r Ú <genexpr>@ s z&_any_pandas_objects.<locals>.<genexpr>)Úany©Útermsr r r Ú_any_pandas_objects< s r. zCallable[[F], F]c s t fdd}|S )Nc sB t | dkrt| d S dd | D }t| s:t| d fS | S )Né r c s s | ] }|j V qd S r( ©r r) r r r r* J s z9_filter_special_cases.<locals>.wrapper.<locals>.<genexpr>)Úlenr r. r )r- Z term_values©Úfr r ÚwrapperD s z&_filter_special_cases.<locals>.wrapper)r )r3 r4 r r2 r Ú_filter_special_casesC s r5 c s dd t D } fdd|D }ddlm} |tt||} | ¡ j}|j}|j}t |}|dk} fdd|D D ]n}
|
t
|
|
|
t } | o| } t |
|
jD ]L\} }| r¼|d |
|
j }}n
|
| | }}|| |¡s|| j|d d
|
||< qq~| ¡ D ]\}}tt||D ]ä\} } | j}t|d rt
|
|t o>|dk}|rR||d n|}t |j| }t |}t tdt|| ¡}|dkrÎ|d krÎd | dt | j d|dd}tj|tt d t |j || dd} | | ¡ q | | jj ¡ qö|t ||fS )Nc S s g | ]\}}t |jd r|qS )r )r r )r! r" r r r r Ú
|
<listcomp>W s z _align_core.<locals>.<listcomp>c s g | ]} | j jqS r )r Úndim©r! r" r, r r r6 X s r )ÚSeriesr/ c 3 s | ]} | j V qd S r( r0 r8 r, r r r* e s z _align_core.<locals>.<genexpr>Úouter)ÚhowÚreindexi' z Alignment difference on axis z. is larger than an order of magnitude on term z, by more than z.4gz; performance may suffer.)ÚcategoryÚ
|
stacklevelF)ÚaxisÚcopy)!r& Zpandasr9 ÚdictÚzipZidxmaxr Z _constructorr r1 r r ÚindexÚis_ÚjoinÚitemsÚranger r Úlog10ÚmaxÚabsÚreprr# ÚwarningsÚwarnr r r r< ÚupdateÚvaluesr )r- Z
|
term_indexZ term_dimsr9 ZndimsZbiggestr r ZnaxesZgt_than_one_axisr Z is_seriesZis_series_and_gt_one_axisr? rF ZaxZitmr" r7 ZtiZ transposeZ reindexerZterm_axis_sizeZreindexer_sizeÚordmÚwr3 r r, r Ú _align_coreU sN
|
|
|
"ÿ ÿrR c C s¦ zt t | ¡} W nT tk
|
rf t| jttfrRt| j}|t || jj
|
f Y S t | j¡df Y S X t dd | D rtdd | D jdfS t| \}}||fS )z
|
Align a set of terms.
|
Nc s s | ] }|j V qd S r( )Z is_scalarr) r r r r* s z align_terms.<locals>.<genexpr>c s s | ] }|j V qd S r( r0 r) r r r r* ¡ s )ÚlistÚcomÚflattenÚ TypeErrorr r r r
|
r r r r Ú result_typeÚallr rR )r- r r r r r Ú align_terms s
|
rY c C sÊ z
|
| j } W n tk
|
r Y nX t |j|¡}t| tsVt| trV| |fd|i|S t |dr| tj
|
kr|tj
|
kr| |¡}nF| | |¡}t |j dkrÆt |dkrÆt|tjsÆt |g¡ |¡}|S )a°
|
Reconstruct an object given its type, raw value, and possibly empty
|
(None) axes.
|
|
Parameters
|
----------
|
typ : object
|
A type
|
obj : object
|
The value to use in the type constructor
|
axes : dict
|
The axes to use to construct the resulting pandas object
|
|
Returns
|
-------
|
ret : typ
|
An object of type ``typ`` with the value `obj` and possible axes
|
`axes`.
|
r r r/ )r ÚAttributeErrorr rW r r r Ú
|
issubclassr r Zbool_Zastyper1 Úshaper Úarray)r Úobjr r Zres_tZ ret_valuer r r Úreconstruct_object¨ s$
|
ÿ
|
þ
|
ýr_ ))Ú__doc__Ú
|
__future__r Ú functoolsr r Útypingr r r rL Únumpyr Z pandas.errorsr Zpandas.util._exceptionsr Zpandas.core.dtypes.genericr
|
r Zpandas.core.baser Zpandas.core.commonÚcoreÚcommonrT Z pandas.core.computation.commonr Zpandas._typingr Zpandas.core.genericr Zpandas.core.indexes.apir r r r. r5 rR rY r_ r r r r Ú<module> s.
|
;
|
