U f/e`@sdZddlmZddlmZmZddlmZmZddl Z ddl Z ddl m Z ddlmZddlmZmZdd lmZddlmmZdd lmZerdd lmZd d ddZddddddZdd ddZddZ e ddZ!ddZ"dd Z#dS)!z! Core eval alignment algorithms. ) annotations)partialwraps) TYPE_CHECKINGSequenceN) FrameOrSeries)PerformanceWarning) ABCDataFrame ABCSeries) PandasObject)result_type_many)Indexz=tuple[partial | type[FrameOrSeries], 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-tiezk1ph/pandas/core/computation/align.py_align_core_single_unary_op"s  rztype[FrameOrSeries]zSequence[Index]zdict[str, Index])rnew_axesrcsfddt|jDS)Ncsi|]\}}||qSrr).0inamerrr 6sz'_zip_axes_from_type..) enumerateZ _AXIS_ORDERS)rrrr!rr3srboolcCstdd|DS)zB Check a sequence of terms for instances of PandasObject. css|]}t|jtVqdSN)rrr rrrrr =sz&_any_pandas_objects..)anytermsrrr_any_pandas_objects9sr+cstfdd}|S)NcsBt|dkrt|dSdd|D}t|s:t|dfS|S)Nrcss|] }|jVqdSr%rr&rrrr'Gsz9_filter_special_cases..wrapper..)lenrr+r )r*Z term_valuesfrrwrapperAs    z&_filter_special_cases..wrapper)r)r0r1rr/r_filter_special_cases@s r2c 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|tddt|j|| dd}||q||jjq|t||fS)NcSs g|]\}}t|jdr|qS)r)rr)rrrrrr Ts z_align_core..csg|]}|jjqSr)rndimrrr)rrr3Usr)Seriesr,c3s|]}|jVqdSr%r-r5r)rrr'bsz_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#Zpandasr6dictzipZidxmaxrZ _constructorrr.rr indexis_joinitemsrangerrlog10maxabsreprr warningswarnrrr9updatevaluesr)r*Z term_indexZ term_dimsr6ZndimsZbiggestrrZnaxesZgt_than_one_axisrZ is_seriesZis_series_and_gt_one_axisr=rDZaxZitmrr4ZtiZ transposeZ reindexerZterm_axis_sizeZreindexer_sizeordmwr0rr)r _align_coreRsF     "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 functoolsrrtypingrrrJZnumpyrZpandas._typingrZ pandas.errorsrZpandas.core.dtypes.genericr r Zpandas.core.baser Zpandas.core.commoncorecommonrRZpandas.core.computation.commonr Zpandas.core.indexes.apir rrr+r2rPrWr]rrrrs*       9