o h@sdZddlmZddlmZmZddlmZmZddl Z ddl Z ddl m Z ddlmZddlmZmZdd lmZddlmmZdd lmZercdd lmZdd lmZdd lm Z ddl!m"Z"d&ddZ#d'ddZ$d(ddZ%d)ddZ&e&d d!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)Indexreturn7tuple[partial | type[NDFrame], dict[str, Index] | None]cCsXd}t|jtjrttj|jjd}||fSt|j}t|jdr(t ||jj }||fS)N)dtypeaxes) isinstancevaluenpndarrayr asanyarrayrtypehasattr_zip_axes_from_typer)termrtyprO/var/www/html/env/lib/python3.10/site-packages/pandas/core/computation/align.py_align_core_single_unary_op's r!r type[NDFrame]new_axesSequence[Index]dict[str, Index]csfddt|jDS)Ncsi|] \}}||qSrr).0inamer#rr :sz'_zip_axes_from_type..) enumerate _AXIS_ORDERS)rr#rr)r r7srboolcCstdd|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=sr4Callable[[F], F]cstfdd}|S)NcsBt|dkr t|dSdd|D}t|st|dfS|S)Nrcs|]}|jVqdSr.rr/rrr r0Kz9_filter_special_cases..wrapper..)lenr!r4r )r3 term_valuesfrr wrapperEs   z&_filter_special_cases..wrapper)r)r=r>rr<r _filter_special_casesDs r?c sddtD}fdd|D}ddlm}|tt||}|j}|j}|j}t |}|dk} fdd|DD]5} t | t } | oI| } t| jD]$\} }| r^|d| j }}n| |}}|| |ss|||||<qOq?|D]\}}tt||D]j\} }|j}t|d rt |t o|dk}|r||dn|}t |j| }t |}ttdt||}|dkr|d krd | d t|jd |dd}tj|ttd|j|| dd}||q||jjqy|t||fS)NcSs g|] \}}t|jdr|qS)r)rr)r&r'rrrr Xs z_align_core..csg|]}|jjqSr)rndimr&r'r2rr r@Ysr)Seriesr6c3s|]}|jVqdSr.r8rBr2rr r0fsz_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+pandasrCdictzipidxmaxr _constructorrr:rr indexis_unionitemsrangerrlog10maxabsreprr(warningswarnrrrDupdatevaluesr)r3 term_index term_dimsrCndimsbiggestrrnaxesgt_than_one_axisr is_seriesis_series_and_gt_one_axisrGrQaxitmr'rAti transpose reindexerterm_axis_sizereindexer_sizeordmwobjrr2r _align_coreVsX       rmcCsz tt|}Wn)ty2t|jttfr(t|j}|t ||jj fYSt |jdfYSwt dd|DrHtdd|DjdfSt|\}}||fS)z Align a set of terms. Ncsr7r.) is_scalarr/rrr r0r9zalign_terms..csr7r.r8r/rrr r0r9)listcomflatten TypeErrorrrr r rrrr result_typeallr rm)r3rrrrr align_termss   rucCsz|j}Wn tyYnwt|j|}t|ts*t|tr*||fd|i|St |dr@|tj kr@|tj kr@||}|S|| |}t |j dkrct |dkrct|tjsct|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`. rrr6)rAttributeErrorrrsrrr issubclassr rbool_astyper:shaperarray)rrlrrres_t ret_valuerrr reconstruct_objects$      r~)rr)rr"r#r$rr%)rr-)rr5)*__doc__ __future__r functoolsrrtypingrrrWnumpyr pandas.errorsrpandas.util._exceptionsrpandas.core.dtypes.genericr r pandas.core.baser pandas.core.commoncorecommonrppandas.core.computation.commonr collections.abcr pandas._typingrpandas.core.genericrpandas.core.indexes.apirr!rr4r?rmrur~rrrr s2              :