hOQ ddlmZddlZddlmZmZmZddlZddl m Z m Z ddl mZddlmZmZddlmZddlmZerdd lmZmZmZdd lmZdd lmZe d- d.d Ze d- d/d Zde j>df d0dZ d1dZ d2 d3dZ!ejDf d4dZ#ejH d5dZ% d6 d5dZ&d7dZ'dZ(ide jRde jTde jVde jXde jZde j\de j^de j`de jbd e jdd!e jfd"e jhd#e jjd$e jld%e jnd&e jpd'e jre'e jtejvejve'e jxejvde'e jhejzejzejze'e jhejzejzejzd(Z>ide j~de jde jde jde jde jde jde jde jd e jd!e jd"e jd#e jd$e jd%e jd&e jd'e je'e jejvejve'e jejvde'e jejzejzejz)e'e jejzejzejz)d(ZRide jde jde jde jde jde jde jde jde jd e jd!e jd"e jd#e jd$e jd%e jd&e jd'e je'e jejvejve'e jejvde'e jejzejzejz)e'e jejzejzejz)d(Zfide jde jde jde jde jde jde jde jde jd e jd!e jd"e jd#e jd$e jd%e jd&e jd'e je'e jejvejve'e jejvde'e jejzejzejz)e'e jejzejzejz)d(Zz d8d*Z{ d9d+Z| d: d;d,Z}y)<) annotationsN) TYPE_CHECKINGcastoverload)algoslib) maybe_promote)ensure_platform_intis_1d_only_ea_dtype)na_value_for_dtype)ensure_wrapped_if_datetimelike) ArrayLikeAxisIntnpt)NDArrayBackedExtensionArray)ExtensionArraycyNarrindexeraxis fill_value allow_fills AD:\jyotish\venv\Lib\site-packages\pandas/core/array_algos/take.pytake_ndr%cyrrrs rrr0rrTc|tjurt|jd}nYtj|jdr9t |j|\}}|j|k7r|j |}t|tjsJt|js!td|}|j||||S|j|||Stj|}t|||||S)a Specialized Cython take which sets NaN values in one pass This dispatches to ``take`` defined on ExtensionArrays. Note: this function assumes that the indexer is a valid(ated) indexer with no out of bound indices. Parameters ---------- arr : np.ndarray or ExtensionArray Input array. indexer : ndarray 1-D array of indices to take, subarrays corresponding to -1 value indices are filed with fill_value axis : int, default 0 Axis to take from fill_value : any, default np.nan Fill value to replace -1 values with allow_fill : bool, default True If False, indexer is assumed to contain no -1 values so no filling will be done. This short-circuits computation of a mask. Result is undefined if allow_fill == False and -1 is present in indexer. Returns ------- subarray : np.ndarray or ExtensionArray May be the same type as the input, or cast to an ndarray. F)compatmMr)rrrrr)r no_defaultr dtype is_np_dtyper astype isinstancenpndarrayr rtakeasarray_take_nd_ndarray)rrrrrr&s rrr;sHS^^#' %@ D ))#))Z@z 99 **U#C c2:: &#399-4c:C88J:D xxJ:xNN **S/C C$ J GGrc"|Ytj|j|tj}|j|jj }}n t |}t||||\}}}d}|jdk(r|jjrd}|r|j}|j|z dz }t|j}t|||<t|} |jjr+||jdz k(rtj| |d} ntj| |} t!|j|j| j||} | ||| ||r | j} | S) Nr&FTF)r&orderr mask_info)r*arangeshapeintpr&typer '_take_preprocess_indexer_and_fill_valuendimflags f_contiguousTlistlentupleempty_get_take_nd_function) rrrrrr&r6 flip_order out_shape_ out_shapeoutfuncs rr.r.xsK))CIIdO277;IIsyy~~'7z%g.#J Wj*$ E:yJ xx1}// eexx$"ciiJ7|Jtj!I yy$#((Q,"6 hhyS9hhy.  #))SYYTY D gsJ'ee Jrcbt|tjs|j|||S|s|j|St |||d|\}}}tj |j |}t|j|j|jd|}||||||S)ao Specialized version for 1D arrays. Differences compared to `take_nd`: - Assumes input array has already been converted to numpy array / EA - Assumes indexer is already guaranteed to be intp dtype ndarray - Only works for 1D arrays To ensure the lowest possible overhead. Note: similarly to `take_nd`, this function assumes that the indexer is a valid(ated) indexer with no out of bound indices. Parameters ---------- arr : np.ndarray or ExtensionArray Input array. indexer : ndarray 1-D array of indices to take (validated indices, intp dtype). fill_value : any, default np.nan Fill value to replace -1 values with allow_fill : bool, default True If False, indexer is assumed to contain no -1 values so no filling will be done. This short-circuits computation of a mask. Result is undefined if allow_fill == False and -1 is present in indexer. mask : np.ndarray, optional, default None If `allow_fill` is True, and the mask (where indexer == -1) is already known, it can be passed to avoid recomputation. r$Tr0rr5) r)r*r+r,r;rCr8rDr<r&) rrrrmaskr&r6rHrIs rtake_1drLsF c2:: &xxJ:xNN xx  #J Wj$$ E:y ((7== .C  #))SYYQ) D gsJ' Jrct|J|dJ|dJ|\}}t|}t|}||f}d}t|j|\}}||jk7r\|dk(}|dk(}|j} |j} ||f| | ff}| s(| s&|j|jj }}t |t |f} t j| |} tj|jj| jjfd} | q|j| jk7rXtj| jj| jjfd} | t| | j} | | ||| || St||| ||| S)zD Specialized Cython take which sets NaN values in one pass. Nrr2r0)rHr)rr6) r r r&anyr:rAr*rC_take_2d_multi_dictgetname_convert_wrapper_take_2d_multi_object)rrrrow_idxcol_idxr6r&row_maskcol_mask row_needs col_needsrGrHrIs r take_2d_multir[s    1: !! ! 1: !! !GW!'*G!'*GwGI&cii<E:  b=b=LLN LLN x(9i*@@ Y!$ 399>>+;:EG c'l*I ((9E *C  " "CIINNCIINN#CT JD | SYY."&&  'GN  #D#))4D  S'sz: J  #*   Jrc|j|jf}|dk(rtj|d}n7|dk(r2|dk(rtj|d}ntj|d}|S|j|jf}|dk(rtj|d}n7|dk(r2|dk(rtj|d}ntj|d}|t ||}|Sy)z Part of _get_take_nd_function below that doesn't need `mask_info` and thus can be cached (mask_info potentially contains a numpy ndarray which is not hashable and thus cannot be used as argument for cached function). r2Nr1r)rR _take_1d_dictrQ_take_2d_axis0_dict_take_2d_axis1_dictrS)r< arr_dtype out_dtypertuprIs r_get_take_nd_function_cachedrcs >>9>> *C qy  d+  19&**35D&**35D  >>9>> *C qy  d+  19&**35D&**35D i0 rcfd}|dkrt|||}|tjfdfd }|S)za Get the appropriate "take" implementation for the given dimension, axis and dtypes. Nr1c>t|}t||||y)N)rrr6)r _take_nd_object)rrrHrrr6s rrIz#_get_take_nd_function..funcWs")'2G Wcy r)returnNone)rcr*nan)r<r`rarr6rIs `` rrDrDEs9 D qy+D)YM |/1vv  KrcNtjf dfd }|S)Nc|j}|j}D|jjdk(r|jd}n|jd}|}||||y)Nmzm8[ns]zM8[ns]r)viewr&kindr()rrrHrr`f fill_wrapras rwrapperz_view_wrapper..wrapperas  ((9%C  ((9%C  $$+'..x8 '..x8 ":.J #w 3rr np.ndarrayrrtrHrtrgrhr*ri)rpr`rarqrrs```` r _view_wrapperrv`s;JL&&4 4",43=4 44$ NrcFtjf dfd }|S)Nchtk(r t|}|j}||||y)Nrm)objectr r()rrrHr conv_dtyperps rrrz!_convert_wrapper..wrapperws3  05Cjj$ #w 3rrsru)rprzrrs`` rrSrSvs6JL&&4 4",43=4 4 Nr)int8r{)r{int32)r{int64)r{float64)int16r)rr|)rr})rr~)r|r|)r|r})r|r~)r}r})r}r~)float32r)rr~)r~r~)ryry))boolr)rry)datetime64[ns]r)timedelta64[ns]r)rqcP||\}}n|dk(}|j}|j|jk7r|j|j}|j|dkDr|j ||||r-t dg|j z}|||<||t|<yy)NrNr)rrH)rOr&r(r8r,slicer<rB) rrrHrrr6rK needs_masking outindexers rrfrfs'm"}  yyCIIjj# yy t-Dk]SXX-  4!+E* rc*|\}}| |\\}}\} } n*|dk(}|dk(}|j} |j} || r |||ddf<| r ||dd|f<t|D]0\} } | dk7s t|D]\} }|dk7s || |f|| | f<2y)NrN)rO enumerate)rrrHrr6rUrVrWrXrYrZiu_jvs rrTrTsGW7@484y)b=b=LLN LLN  )C!  )C8 7#2 8!'*17 #BE C1I+$rcTd}|s)|j|jj}}d}nwt|j|\}}||jk7rO|d}n|dk(}t|j }||f}|s&|j|jj}}|||fS)N)NFTrN)r&r:r rrO)rrrrrKr6r&rs rr;r;4s8rs "  2:C G6                  ~~ :H :H :H  :H  :Hz. . (. .  .  .h)- 7 7 !7 7 ' 7  7zvv9 9 >9 9x " """/7"?F""R    6, h00x22x2266  44  44 44(884444(8844(88H<<H<< H<$>"((S%h&B&BBHHdS*7$$bhh"((+-:$$bhh"((-/ 8h66x88x88<<  ::  :: ::(>>::::(>>::(>>HBBHBB HBB!"(>>#$$(("((BHH&**BHHd+8**BHHbhh"((+-:**BHHbhh"((-7@h66x88x88<<  ::  :: ::(>>::::(>>::(>>HBBHBB HBB!"(>>#$$(("((BHH&**BHHd+8**BHHbhh"((+-:**BHHbhh"((-7@h66x88x88<<  ::  :: ::(>>::::(>>::(>>HBBHBB HBB!"(>>#$$(("((BHH&**BHHd+8**BHHbhh"((+-:**BHHbhh"((-7B, , !, ,  , ,.+ + >+ +  +D*. ( ( !( ( ' (r