U c @sdZddlmZddlmZmZddlmZmZm Z ddl Z ddl Z ddl mZddlmZddlmZmZdd lmZddlmmZdd lmZerdd lmZdd lmZdd l m!Z!ddddZ"ddddddZ#ddddZ$ddddZ%e%ddZ&d d!Z'd"d#Z(dS)$z! Core eval alignment algorithms. ) annotations)partialwraps) TYPE_CHECKINGCallableSequenceN)PerformanceWarning)find_stack_level) ABCDataFrame ABCSeries) PandasObject)result_type_many)F)NDFrame)Indexz7tuple[partial | type[NDFrame], dict[str, Index] | None])returncCsRd}t|jtjr&ttj|jjd}n$t|j}t|jdrJt ||jj }||fS)N)dtypeaxes) isinstancevaluenpndarrayrZ asanyarrayrtypehasattr_zip_axes_from_typer)termrtyprA/tmp/pip-unpacked-wheel-g7fro6k3/pandas/core/computation/align.py_align_core_single_unary_op&s  rz type[NDFrame]zSequence[Index]zdict[str, Index])rnew_axesrcsfddt|jDS)Ncsi|]\}}||qSrr).0inamer rr :sz'_zip_axes_from_type..) enumerateZ _AXIS_ORDERS)rr rr$rr7srboolcCstdd|DS)zB Check a sequence of terms for instances of PandasObject. css|]}t|jtVqdSN)rrr r!rrrr Asz&_any_pandas_objects..)anytermsrrr_any_pandas_objects=sr.zCallable[[F], F]cstfdd}|S)NcsBt|dkrt|dSdd|D}t|s:t|dfS|S)Nrcss|] }|jVqdSr(rr)rrrr*Ksz9_filter_special_cases..wrapper..)lenrr.r )r-Z term_valuesfrrwrapperEs    z&_filter_special_cases..wrapper)r)r3r4rr2r_filter_special_casesDs r5c sddtD}fdd|D}ddlm}|tt||}|j}|j}|j}t |}|dk} fdd|DD]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||dn|}t |j| }t |}ttdt||}|dkr|d krd | dt|jd|dd}tj|ttdt|j|| dd}||q||jjq|t ||fS)NcSs g|]\}}t|jdr|qS)r)rr)r!r"rrrr Xs z_align_core..csg|]}|jjqSr)rndimr!r"r,rrr6Ysr)Seriesr/c3s|]}|jVqdSr(r0r8r,rrr*fsz_align_core..outer)howreindexi'zAlignment difference on axis z. is larger than an order of magnitude on term z, by more than z.4gz; performance may suffer.)category stacklevelF)axiscopy)!r&Zpandasr9dictzipZidxmaxrZ _constructorrr1rr indexis_joinitemsrangerrlog10maxabsreprr#warningswarnrr rr<updatevaluesr)r-Z term_indexZ term_dimsr9ZndimsZbiggestrrZnaxesZgt_than_one_axisrZ is_seriesZis_series_and_gt_one_axisr?rFZaxZitmr"r7ZtiZ transposeZ reindexerZterm_axis_sizeZreindexer_sizeordmwr3rr,r _align_coreVsN     "rRcCsztt|}WnTtk rft|jttfrRt|j}|t ||jj fYSt |jdfYSXt dd|Drtdd|DjdfSt|\}}||fS)z Align a set of terms. Ncss|] }|jVqdSr()Z is_scalarr)rrrr*szalign_terms..css|] }|jVqdSr(r0r)rrrr*s)listcomflatten TypeErrorrrr r rrrr result_typeallr rR)r-rrrrr align_termss  rYcCsz |j}Wntk rYnXt|j|}t|tsVt|trV||fd|i|St |dr|tj kr|tj kr||}nF|| |}t |j dkrt |dkrt|tjst|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`. rrr/)rAttributeErrorrrWrrr issubclassr rZbool_Zastyper1shaperarray)robjrrZres_tZ ret_valuerrrreconstruct_objects$     r_))__doc__ __future__r functoolsrrtypingrrrrLZnumpyrZ pandas.errorsrZpandas.util._exceptionsr Zpandas.core.dtypes.genericr r Zpandas.core.baser Zpandas.core.commoncorecommonrTZpandas.core.computation.commonr Zpandas._typingrZpandas.core.genericrZpandas.core.indexes.apirrrr.r5rRrYr_rrrrs.         ;