*j[CddlmZddlmZddlmZmZmZmZm Z m Z ddl Z ddl mZddlmZddlmZddlmZmZmZmZmZmZmZmZmZmZmZmZdd l m!Z!dd l"m#Z#dd l$m%Z%m&Z&dd l'm(Z(dd l)m*Z*m+Z+m,Z,ddl-m.Z.ddl/m0Z0ddl1m2Z2m3Z3m4Z5ddl6m7Z7ddl8m9Z9ddl:m;Z;ddlm?Z?m@Z@ddlAmBZBerddlCmDZDddlmEZEmFZFddlGmHZHddZIGddee;ZJy)) annotations)wraps) TYPE_CHECKINGAnyLiteralSelfcastoverloadN)lib) NDArrayBacked)is_supported_dtype) ArrayLikeAxisIntDtypeF FillnaOptionsPositionalIndexer2DPositionalIndexerTuple ScalarIndexerSequenceIndexerShape TakeIndexernptAbstractMethodError)doc)validate_bool_kwargvalidate_insert_loc) pandas_dtype)DatetimeTZDtypeExtensionDtype PeriodDtype)array_equivalent)missing)takeuniquevalue_counts_internal)quantile_with_mask)shift)ExtensionArray) extract_array)check_array_indexergetitem_returns_view) nargminmax)Sequence) NumpySorterNumpyValueArrayLike)SeriescJtfd}tt|S)z~ Decorator to ravel a 2D array before passing it to a cython operation, then reshape the result to our own shape. c|jdk(r |g|i|S|jj}|jd}|g|i|}|jrdnd}|j |j |S)NKrC)order)ndim_ndarrayflagsravel f_contiguousreshapeshape)selfargskwargsr;flatresultr8meths .methodPsy 99>.t.v. . ##zz#d,T,V,))s~~djj~66)rr r)rErGs` rF ravel_compatrIJs)  4[77 6?rHceZdZUdZded<ded<dZdZed,dZed-d.d Zd/d.d Zd d d d d0dZ d1dZ e dZ d2dZ dZ d3dZd4dZd5d6dZd5d6dZd,dZe eej, d7 d8fd Zeej. d9 d:dZeej0d;ddZe d?d Z d@d!Zd d dd" dAd#Zeej:dBdCd$ZdDd%ZdEd&ZdFd'Z dGd(Z!dHdId)Z" dJd*Z#e dKd+Z$xZ%S)LNDArrayBackedExtensionArrayzB ExtensionArray that is backed by a single NumPy ndarray. np.ndarrayr:r_internal_fill_valuec|S)zA Wrap numpy type in our dtype.type if necessary. )r@xs rF _box_funcz%NDArrayBackedExtensionArray._box_funcks rHct|Nrr@values rF_validate_scalarz,NDArrayBackedExtensionArray._validate_scalarqs !$''rHcyrSrOr@s rFviewz NDArrayBackedExtensionArray.viewwsrHcyrSrO)r@dtypes rFrYz NDArrayBackedExtensionArray.viewzsddlm}|j}|||||S)Nr) hash_array)encodinghash_key categorize)pandas.core.util.hashingrr:)r@rrrrr~s rF_hash_pandas_objectz/NDArrayBackedExtensionArray._hash_pandas_objects& 8 XZ  rHcftj|t}tj|dS)Nr^T)convert_non_numeric)npasarrayobjectr maybe_convert_objects)r@r~s rF_cast_pointwise_resultz2NDArrayBackedExtensionArray._cast_pointwise_results%F&1((TJJrHTcht|d|s|jr tdt|d|S)Nskipna)Encountered an NA value with skipna=Falseargminrtr_hasna ValueErrorr.r@rtrs rFrz"NDArrayBackedExtensionArray.argmin0FH-$++HI I$t44rHcht|d|s|jr tdt|d|S)Nrrargmaxrrrs rFrz"NDArrayBackedExtensionArray.argmaxrrHcNt|j}|j|SrS)r&r:rd)r@rws rFr&z"NDArrayBackedExtensionArray.uniques!$--(&&x00rHctj|Dcgc]}|jc}s.|Dchc]}t|j}}t d|t |||Scc}wcc}w)Nz"to_concat must have the same dtyper)r dtypes_all_equalr[strrsuper_concat_same_type)rm to_concatrtrPdtypes __class__s rFrz-NDArrayBackedExtensionArray._concat_same_typesn##i$@iQWWi$@A,56Iqc!''lIF6A6J Jw((>> %A6s A.A3c`|j|}|jj|||S)N)sidesorter)_validate_setitem_valuer: searchsorted)r@rUrrnpvalues rFrz(NDArrayBackedExtensionArray.searchsorteds0..u5}}))'V)LLrHczd}|j|}t|j|||}|j|S)Nr)rVr)r:rd)r@periodsrsrt new_valuess rFr)z!NDArrayBackedExtensionArray.shifts>**:6 4=='4D &&z22rHc|jr tdt||}|j|}||j|<y)NzCannot modify read-only array) _readonlyrr,rr:)r@keyrUs rF __setitem__z'NDArrayBackedExtensionArray.__setitem__ s? >><= =!$,,,U3" crHc|SrSrOrTs rFrz3NDArrayBackedExtensionArray._validate_setitem_values rHcyrSrOr@rs rF __getitem__z'NDArrayBackedExtensionArray.__getitem__s69rHcyrSrOrs rFrz'NDArrayBackedExtensionArray.__getitem__srHctj|r_|j|}|jdk(r|j |S|j |}t ||r|j|_|St|d}t||}|j|}tj|r|j |S|j |}t ||r|j|_|S)Nr5T) extract_numpy) r is_integerr:r9rQrdr-rr+r, is_scalar)r@rrDs rFrz'NDArrayBackedExtensionArray.__getitem__s >># ]]3'FyyA~~~f--,,V4F#D#.#'>> MCt4!$,s# == >>&) )((0 c *#~~F  rH)limit limit_areacopycz|j}|jrtj||j}|j j }|r|j}|||||j |j }|r|j|}|S|}|S|r|j}|S|}|S)N)r9)rrmask) isnaanyr$ get_fill_funcr9r:Trrd) r@rGrrrrfuncnpvaluesrs rF_pad_or_backfillz,NDArrayBackedExtensionArray._pad_or_backfill:syy{ 88:((dii@D}}H#==? :DFF KzzH!44X>  "  JJrHc|j}|F|t|kr8|j|kD}|jr|j }d||<t j ||t|}|jr|r|j }n|dd}|||<|S|j||s|dd}|S|j }|S)NF)rlencumsumrrr$check_value_sizer)r@rUrrrmodifyrs rFfillnaz"NDArrayBackedExtensionArray.fillnaXsyy{  T!2[[]U*Fzz|yy{$V ((   I  88:!YY[ !!W $Jt   ( ( /!!W "YY[ rHch||jdk(r|j|S|j|S)Nr5)r9rQrd)r@rtrDs rF_wrap_reduction_resultz2NDArrayBackedExtensionArray._wrap_reduction_result~s1 <499>>>&) )&&v..rHch|j|}tj|j||y)a  Analogue to np.putmask(self, mask, value) Parameters ---------- mask : np.ndarray[bool] value : scalar or listlike Raises ------ TypeError If value cannot be cast to self.dtype. N)rrputmaskr:)r@rrUs rF_putmaskz$NDArrayBackedExtensionArray._putmasks(,,U3 4==$.rHc|j|}tj||j|}|j|jjk7r t d|j |S)a Analogue to np.where(mask, self, value) Parameters ---------- mask : np.ndarray[bool] value : scalar or listlike Raises ------ TypeError If value cannot be cast to self.dtype. zNSomething has gone wrong, please report a bug at github.com/pandas-dev/pandas/)rrwherer:r[AssertionErrorrd)r@rrU res_valuess rF_wherez"NDArrayBackedExtensionArray._wheresg,,U3XXdDMM59   t}}22 2 0  &&z22rHc(t|t|}|j|}tj|j d|tj |g|j j|j |df}|j|S)a Make new ExtensionArray inserting new item at location. Follows Python list.append semantics for negative values. Parameters ---------- loc : int item : object Returns ------- type(self) Nr^) rrrVr concatenater:rr[rd)r@locitemcodenew_valss rFinsertz"NDArrayBackedExtensionArray.inserts"#s4y1$$T*>> ds# D6)<)<= cd#  &&x00rHc|jdk7rtddlm}m}|r||j j }n |j }t|d|}|jtj|jj}|||jjd}||j||jdS)z Return a Series containing counts of unique values. Parameters ---------- dropna : bool, default True Don't include counts of NA values. Returns ------- Series r5r)Indexr2F)sortdropna)namer)indexrr)r9NotImplementedErrorpandasrr2rr: value_countsrdrrr_datar_values)r@rrr2r~rD index_arrrs rFrz(NDArrayBackedExtensionArray.value_countss 99>% % 499;,'00F]]Ff5@++BJJv||7I7I,JK ifll&7&7eDfnnE %PPrHc tj|j}|j}|j}t |||||}|j |jj k(r|j|St||SrS) rrrr:rMr(r[rdrf)r@qs interpolationrrlrsrs rF _quantilez%NDArrayBackedExtensionArray._quantileszzz$))+&mm.. 'T:r=Q   t}}22 2**:6 6 4:j) )rHc|jg|}tj||jj}|j |S)z Analogous to np.empty(shape, dtype=dtype) Parameters ---------- shape : tuple[int] dtype : ExtensionDtype r^)_from_sequenceremptyr:r[rd)rmr?r[rlbackings rF_emptyz"NDArrayBackedExtensionArray._empty sD  5 1((5 (:(:;%%g..rH)returnr).)r[z Dtype | NonerrrS) rvrrrrzrsrrtrrr)rrz)rrL)rrrrrrzrznpt.NDArray[np.uint64])r~rrr)rT)rtrrrz)r)rzSequence[Self]rtrrr)leftN)rUz$NumpyValueArrayLike | ExtensionArrayrzLiteral['left', 'right']rzNumpySorter | Nonerznpt.NDArray[np.intp] | np.intp)r5N)rintrr)rNone)rrrr)rz(SequenceIndexer | PositionalIndexerTuplerr)rrrz Self | Any) rGrr int | Nonerz#Literal['inside', 'outside'] | Nonerrzrr)NT)rrrrzrr)rtzAxisInt | Nonerr)rnpt.NDArray[np.bool_]rr)r@rrrrr)rrrr)T)rrzrr2)rznpt.NDArray[np.float64]rrrr)r?rr[r!rr)&__name__ __module__ __qualname____doc____annotations__rQrVr rYr%r| classmethodrrrrrrrr&rr*rrr)rrrrrrrrrrrr __classcell__)rs@rFrKrK^s  (  ??"%P! 11 1  1  1 1,W338  *- ;?  K 551 ) )* ?! ? ?  ?+ ?  $ $%*0%) M3M'M# M ( M&M   33#99  5    >!:>   8    <      J//$381@QB* #**  *0 / /rHrK)rErrr)K __future__r functoolsrtypingrrrrr r numpyr pandas._libsr pandas._libs.arraysr pandas._libs.tslibsr pandas._typingrrrrrrrrrrrr pandas.errorsrpandas.util._decoratorsrpandas.util._validatorsrrpandas.core.dtypes.commonrpandas.core.dtypes.dtypesr r!r"pandas.core.dtypes.missingr# pandas.corer$pandas.core.algorithmsr%r&r'r pandas.core.array_algos.quantiler("pandas.core.array_algos.transformsr)pandas.core.arrays.baser*pandas.core.constructionr+pandas.core.indexersr,r-pandas.core.sortingr.collections.abcr/r0r1rr2rIrKrOrHrFrs"-2    .' 3 8 @422+( (}/-}/rH