U Mf@sdZddlmZddlmZmZddlmZmZddl Z ddl Z ddl m Z ddlmZddlmZmZdd lmZddlmmZdd lmZerdd lmZdd lmZd dddZddddddZ ddddZ!ddZ"e"ddZ#ddZ$d d!Z%dS)"z! Core eval alignment algorithms. ) annotations)partialwraps) TYPE_CHECKINGSequenceN)PerformanceWarning)find_stack_level) ABCDataFrame ABCSeries) PandasObject)result_type_many)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-eb6vo0j3/pandas/core/computation/align.py_align_core_single_unary_op#s  rz type[NDFrame]zSequence[Index]zdict[str, Index])rnew_axesrcsfddt|jDS)Ncsi|]\}}||qSrr).0inamerrr 7sz'_zip_axes_from_type..) enumerateZ _AXIS_ORDERS)rrrr"rr4srboolcCstdd|DS)zB Check a sequence of terms for instances of PandasObject. css|]}t|jtVqdSN)rrr rrrrr >sz&_any_pandas_objects..)anytermsrrr_any_pandas_objects:sr,cstfdd}|S)NcsBt|dkrt|dSdd|D}t|s:t|dfS|S)Nrcss|] }|jVqdSr&rr'rrrr(Hsz9_filter_special_cases..wrapper..)lenrr,r )r+Z term_valuesfrrwrapperBs    z&_filter_special_cases..wrapper)r)r1r2rr0r_filter_special_casesAs r3c 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)rr rrrr Us z_align_core..csg|]}|jjqSr)rndimrr r*rrr4Vsr)Seriesr-c3s|]}|jVqdSr&r.r6r*rrr(csz_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$Zpandasr7dictzipZidxmaxrZ _constructorrr/rr indexis_joinitemsrangerrlog10maxabsreprr!warningswarnrrrr:updatevaluesr)r+Z term_indexZ term_dimsr7ZndimsZbiggestrrZnaxesZgt_than_one_axisrZ is_seriesZis_series_and_gt_one_axisr=rDZaxZitmr r5ZtiZ transposeZ reindexerZterm_axis_sizeZreindexer_sizeordmwr1rr*r _align_coreSsN     "rPcCsztt|}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&r.r'rrrr(s)listcomflatten TypeErrorrrr r rrrr result_typeallr rP)r+rrrrr align_termss  rWcCsz |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-)rAttributeErrorrrUrrr issubclassr rZbool_Zastyper/shaperarray)robjrrZres_tZ ret_valuerrrreconstruct_objects$     r])&__doc__ __future__r functoolsrrtypingrrrJZnumpyrZ pandas.errorsrZpandas.util._exceptionsrZpandas.core.dtypes.genericr r Zpandas.core.baser Zpandas.core.commoncorecommonrRZpandas.core.computation.commonr Zpandas.core.genericr Zpandas.core.indexes.apirrrr,r3rPrWr]rrrrs,        ;