a Of@s"dZddlmZddlmZmZddlmZmZddl Z ddl Z ddl m Z ddlmZddlmZmZdd lmZddlmmZdd lmZerdd lmZdd lmZdd lm Z ddl!m"Z"ddddZ#ddddddZ$ddddZ%ddddZ&e&dd Z'd!d"Z(d#d$Z)dS)%z! Core eval alignment algorithms. ) annotations)partialwraps) TYPE_CHECKINGCallableN)PerformanceWarning)find_stack_level) ABCDataFrame ABCSeries) PandasObject)result_type_many)Sequence)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)termrtyprV/var/www/ai-form-bot/venv/lib/python3.9/site-packages/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 :z'_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 Ar&z&_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+Kr&z9_filter_special_cases..wrapper..)lenrr/r )r.Z term_valuesfrrwrapperEs    z&_filter_special_cases..wrapper)r)r4r5rr3r_filter_special_casesDs r6c sddtD}fdd|D}ddlm}|tt||}|j}|j}|j}t |}|dk} fdd|DD]j} t | t } | o| } t| jD]H\} }| r|d| j }}n | |}}|| |s|||||<qq~|D]\}}tt||D]\} }|j}t|d r t |t o:|dk}|rN||dn|}t |j| }t |}ttdt||}|dkr|d krd | d t|jd |dd}tj|ttd|j|| dd}||q ||jjq|t||fS)NcSs g|]\}}t|jdr|qS)r)rr)r!r"rrrr Xr&z_align_core..csg|]}|jjqSr)rndimr!r"r-rrr7Yr&r)Seriesr0c3s|]}|jVqdSr)r1r9r-rrr+fr&z_align_core..reindexi'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'Zpandasr:dictzipZidxmaxrZ _constructorrr2rr indexis_unionitemsrangerrlog10maxabsreprr#warningswarnrrr;updatevaluesr)r.Z term_indexZ term_dimsr:ZndimsZbiggestrrZnaxesZgt_than_one_axisrZ is_seriesZis_series_and_gt_one_axisr>rEaxZitmr"r8tiZ transposeZ reindexerZterm_axis_sizeZreindexer_sizeordmwobjrr-r _align_coreVsR      rTcCsztt|}WnRtydt|jttfrPt|j}|t ||jj fYSt |jdfYS0t dd|Drtdd|DjdfSt|\}}||fS)z Align a set of terms. Ncss|] }|jVqdSr))Z is_scalarr*rrrr+r&zalign_terms..css|] }|jVqdSr)r1r*rrrr+r&)listcomflatten TypeErrorrrr r rrrr result_typeallr rT)r.rrrrr align_termss   r[cCsz |j}WntyYn0t|j|}t|tsTt|trT||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`. rrr0)rAttributeErrorrrYrrr issubclassr rZbool_Zastyper2shaperarray)rrSrrZres_tZ ret_valuerrrreconstruct_objects$      r`)*__doc__ __future__r functoolsrrtypingrrrKnumpyrZ pandas.errorsrZpandas.util._exceptionsrZpandas.core.dtypes.genericr r Zpandas.core.baser Zpandas.core.commoncorecommonrVZpandas.core.computation.commonr collections.abcr Zpandas._typingrZpandas.core.genericrZpandas.core.indexes.apirrrr/r6rTr[r`rrrrs0          :