
    PiA<                      d dl mZ d dlZd dlZd dlmZmZmZmZm	Z	m
Z
 d dlZd dlZd dlmZmZmZ d dlmZmZ d dlmZ d dlmZmZmZ d dlmZ d d	lmZ d d
l m!Z! d dl"m#Z#m$Z$ d dl%m&Z&m'Z'm(Z(m)Z)m*Z* d dl+m,Z,m-Z-m.Z.m/Z/m0Z0m1Z1 d dl2m3Z3m4Z4m5Z5m6Z6m7Z7 d dl8m9Z9m:Z:m;Z;m<Z< d dl=m>Z> d dl?m@Z@mAZAmBZB d dlCmDZD d dlEmFc mGZH d dlImJZJmKZKmLZLmMZMmNZN d dlOmPZP d dlQmRZRmSZSmTZT d dlUmVZV d dlWmXZXmYZYmZZZm[Z[m\Z\m]Z] d dl^m_Z_ d dl`maZa d dlbmFc mcZd d dlemfZf d dlgmhZhmiZi d dljmkZk d dllmmZm er0d dlnmoZompZpmqZqmrZr d dlsmtZtmuZumvZvmwZwmxZxmyZymzZzm{Z{m|Z|m}Z} d d l~mZ d d!lmZ  ej        d"          Z G d# d$eaej                  Z G d% d&e          Z G d' d(e          Z G d) d*e          Z G d+ d,e          Z G d- d.e          ZdJd2ZdKd6Z	 dLdMd;Zdd<dNd?ZdOdAZdPdDZdLdQdFZdRdSdHZdJdIZdS )T    )annotationsN)TYPE_CHECKINGAnyLiteralSelfcastfinal)NaT	internalslib)BlockPlacementBlockValuesRefs)NA)AbstractMethodErrorOutOfBoundsDatetimePandas4Warning)cache_readonly)find_stack_level)validate_bool_kwarg)astype_array_safeastype_is_view)LossySetitemErrorcan_hold_elementconvert_dtypesfind_result_typenp_can_hold_element)is_1d_only_ea_dtypeis_float_dtypeis_integer_dtypeis_list_like	is_scalaris_string_dtype)DatetimeTZDtypeExtensionDtypeIntervalDtypeNumpyEADtypePeriodDtype)ABCDataFrameABCIndexABCNumpyExtensionArray	ABCSeries)is_re)is_valid_na_for_dtypeisnana_value_for_dtype)missing)extract_bool_arrayputmask_inplaceputmask_without_repeatsetitem_datetimelike_compatvalidate_putmask)quantile_compat)compare_or_regex_searchreplace_regexshould_use_regex)shift)DatetimeArrayExtensionArrayIntervalArrayNumpyExtensionArrayPeriodArrayTimedeltaArray)StringDtype)PandasObject)expressions)ensure_wrapped_if_datetimelikeextract_array)check_setitem_lengths)get_values_for_csv)Callable	GeneratorIterableSequence)
	ArrayLikeAxisIntDtypeBackendDtypeObjFillnaOptionsIgnoreRaiseInterpolateOptionsQuantileInterpolationShapenpt)Index)NDArrayBackedExtensionArrayobjectc                     e Zd ZU dZded<   ded<   ded<   ded	<   d
ZdZeedd                        Z	eedd                        Z
eedd                        Zeedd                        Zeed                         Zeedd                        Zeedd                        Zed             Zeed                         Zed             Zedd            Zej        dd            Ze	 	 ddd!            Ze	 	 ddd#            Zedd%            Zedd&            Zedd)            Zedd,            Ze	 ddd/            Zedd2            Zedd5            Zedd7            Zedd8            Z edd:            Z!edd<            Z"edd=            Z#edd?            Z$edd@            Z%	 	 	 	 	 	 dddJZ&eeddL                        Z'e	 	 dddR            Z(edSddTddV            Z)eddX            Z*dddZZ+edd[            Z,e	 	 ddd^            Z-e	 	 ddd_            Z.e	 	 ddde            Z/e	 	 dddg            Z0ddjZ1dk Z2eddm            Z3ddpZ4ddrZ5dddtZ6ede7j8        fddx            Z9dd{Z:dd|Z;dd}Z<dd~Z=	 	 dddZ>ddddddZ?eddddddd            Z@edd            ZAdddZBe	 ddd            ZCedd            ZDddZEedd            ZFedd            ZGdddZHdS )Blockz
    Canonical n-dimensional unit of homogeneous dtype contained in a pandas
    data structure

    Index-ignorant; let the container take care of that
    znp.ndarray | ExtensionArrayvaluesintndimr   refsrH   __init__ Freturnboolc                ,    t          | j                   S )z
        We validate dimension for blocks that can hold 2D values, which for now
        means numpy dtypes or EA dtypes like DatetimeTZDtype and PeriodDtype.
        )r   dtypeselfs    p/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/pandas/core/internals/blocks.py_validate_ndimzBlock._validate_ndim   s     'tz2222    c                ,    | j         j        t          k    S N)r[   rd   
_dtype_objre   s    rg   	is_objectzBlock.is_object   s     { J..ri   c                @    t          j        | j        j                   S rk   )r   is_np_dtyper[   rd   re   s    rg   is_extensionzBlock.is_extension   s     ?4;#45555ri   c                    | j          S rk   )rp   re   s    rg   _can_consolidatezBlock._can_consolidate   s     $$$ri   c                (    | j         | j        j        fS rk   )rr   rd   namere   s    rg   _consolidate_keyzBlock._consolidate_key   s     $djo55ri   c                d    | j         }t          |t          j                   r	|j        dvS |j        S )z7
        Can we store NA values in this Block?
        iub)rd   
isinstancenpkind_can_hold_narf   rd   s     rg   r{   zBlock._can_hold_na   s5     
eRX&& 	+:U**!!ri   c                P    | j         j        t          j        t                    k    S )z^
        We can be bool if a) we are bool dtype or b) object dtype with bool objects.
        )r[   rd   ry   rb   re   s    rg   is_boolzBlock.is_bool   s     { BHTNN22ri   c                *    t          | j                  S rk   )external_valuesr[   re   s    rg   r   zBlock.external_values   s    t{+++ri   c                .    t          | j        d          S )NF)compat)r/   rd   re   s    rg   
fill_valuezBlock.fill_value   s     "$*U;;;;ri   c                ^    | j         t          k    rt          || j                   r| j        }|S rk   )rd   rl   r-   r   rf   values     rg   _standardize_fill_valuezBlock._standardize_fill_value   s/     :##(=eTZ(P(P#OEri   r   c                    | j         S rk   	_mgr_locsre   s    rg   mgr_locszBlock.mgr_locs   s
    ~ri   new_mgr_locsNonec                    || _         d S rk   r   )rf   r   s     rg   r   zBlock.mgr_locs   s    %ri   N	placementBlockPlacement | NoneBlockValuesRefs | Nonec                ~    || j         }| j        rt          || j                  }t	          ||| j        |          S )zm
        Create a new block, with type inference propagate any values that are
        not specified
        Nr]   r   r]   r^   )r   rp   ensure_block_shaper]   	new_blockrf   r[   r   r^   s       rg   
make_blockzBlock.make_block   sJ     I 	@'TY???F9494PPPPri   r   c                V    || j         } t          |           ||| j        |          S )z2Wrap given values in a block of same type as self.Nr   )r   typer]   r   s       rg   make_block_same_classzBlock.make_block_same_class   s4     I tDzz&IDIDQQQQri   strc                    t          |           j        }| j        dk    r| dt          |            d| j         }n@d                    d | j        D                       }| d| j        j         d| d| j         }|S )N   z: z dtype: z x c                ,    g | ]}t          |          S r`   )r   ).0ss     rg   
<listcomp>z"Block.__repr__.<locals>.<listcomp>  s    ;;;1A;;;ri   z, z	, dtype: )	r   __name__r]   lenrd   joinshaper   indexer)rf   rt   resultr   s       rg   __repr__zBlock.__repr__  s     Dzz"9>>??D		??4:??FFJJ;;
;;;<<EUU 5UUUUUUFri   c                *    t          | j                  S rk   )r   r[   re   s    rg   __len__zBlock.__len__  s    4;ri   slcslicec                    | j         |         }|                     |          }| j        } t          |           ||| j        |          S )zC
        Perform __getitem__-like, return result as block.
        r^   )r   _slicer^   r   r]   )rf   r   r   
new_valuesr^   s        rg   slice_block_columnszBlock.slice_block_columns  sI    
 ~c*[[%%
ytDzz*lDIDIIIIri   indicesnpt.NDArray[np.intp]c                    | j         |         }|                     |          } t          |           ||| j        d          S z
        Perform __getitem__-like, return result as block.

        Only supports slices that preserve dimensionality.
        Nr   )r   r   r   r]   )rf   r   r   r   s       rg   take_block_columnszBlock.take_block_columns!  sB     ~g.[[))
tDzz*lDIDIIIIri   slicerref_inplace_opc                    |                      |          }|r| j                                        r| j        nd} t          |           ||| j        |          S r   )r   r^   has_referencer   r]   )rf   r   r   r   r   r^   s         rg   getitem_block_columnszBlock.getitem_block_columns0  sZ     [[((
 .U$)2I2I2K2KUtyyQUtDzz*lDIDIIIIri   elementr   c                N    t          |d          }t          | j        |          S )z#require the same dtype as ourselvesTextract_numpy)rE   r   r[   )rf   r   s     rg   _can_hold_elementzBlock._can_hold_element=  s(      t<<<W555ri   r   rL   c                "    |j         | j         k    S )z
        Should we set self.values[indexer] = value inplace or do we need to cast?

        Parameters
        ----------
        value : np.ndarray or ExtensionArray

        Returns
        -------
        bool
        rd   r   s     rg   should_storezBlock.should_storeC  s     {dj((ri   list[Block]c                f     || j         fi |}t          |          }|                     |          S )z[
        apply the function to my values; return a block if we are not
        one
        )r[   maybe_coerce_values_split_op_result)rf   funckwargsr   s       rg   applyzBlock.applyU  s>     dk,,V,,$V,,$$V,,,ri   c                    | j         dk    sJ  || j                  }| j        j         dk    r|}n|                    dd          }|                     |          S )N   r   )r]   r[   reshaper   )rf   r   r   
res_valuess       rg   reducezBlock.reduce`  s`     yA~~~~dk"";q  JJA..Jz***ri   r   c                   |j         dk    rt          |j        t                    rg }t	          | j                  D ]j\  }}t          |j                  s|||dz            }n||         }t          |          }|                     ||          }|	                    |           k|S |                     |          }|gS )Nr   )r[   r   )
r]   rx   rd   r$   	enumerater   r   r   r   append)	rf   r   nbsilocvalsbpblocknbs	            rg   r   zBlock._split_op_resulto  s     ;??z&,GG? C#DN33 " "3*6<88 %!!a!e),DD!!9D#C((trBB

5!!!!J__V$$tri   Generator[Block]c              #     K   | j         dk    sJ t          | j                  D ]W\  }}| j        t	          ||dz                      }t          |          } t          |           ||d| j                  }|V  XdS )zD
        Split a block into a list of single-column blocks.
        r   r   r   N)r]   r   r   r[   r   r   r   r^   )rf   r   ref_locr   r   r   s         rg   _splitzBlock._split  s      
 yA~~~~#DN33 	 	JAw;uQA/D((BdDBQTYGGGBHHHH	 	ri   c                    | j         dk    r| j        d         dk    sJ g }|                                 D ]#} ||g|R i |}|                    |           $|S )z
        Split the block and apply func column-by-column.

        Parameters
        ----------
        func : Block method
        *args
        **kwargs

        Returns
        -------
        List[Block]
        r   r   r   )r]   r   r   extend)rf   r   argsr   
res_blocksr   rbss          rg   split_and_operatezBlock.split_and_operate  s}     yA~~$*Q-1"4"4"44
++-- 	# 	#B$r+D+++F++Cc""""ri   raise_on_upcastc                Z   t          | j        j        |          }|| j        k    rt          d          t	          |          rnt          | j        j                  rUt          |          rF|t          ur=t          |t          j
        t          j        f          rt          j        |          sd}nht          |t          j                  rN|j        dk    rCt          | j        j                  r*t          |j                  rt!          j        |          rd}|r t%          d| d| j        j         d          | j        j        |k    r t          d| d| j        j         d	          	 |                     |          S # t(          $ r}t)          d
| j         d          |d}~ww xY w)z
        coerce the current block to a dtype compat for other
        we will return a block, possibly object, and not raise

        we can also safely try to coerce to the same dtype
        and will receive the same block
        z\Something has gone wrong, please report a bug at https://github.com/pandas-dev/pandas/issuesFr   zInvalid value 'z' for dtype ''zDid not expect new dtype z to equal self.dtype zE. Please report a bug at https://github.com/pandas-dev/pandas/issues.z0Incompatible (high-resolution) value for dtype='z$'. Explicitly cast before operating.N)r   r[   rd   AssertionErrorr!   r   r.   r
   rx   ry   
datetime64timedelta64isnatndarrayr]   r   r   has_only_ints_or_nan	TypeErrorastyper   )rf   otherr   	new_dtypeerrs        rg   coerce_to_target_dtypezBlock.coerce_to_target_dtype  s    %T[%6>>	
"" >   e	$ !233	$ U	$ S  52="."ABB !GIxPU !
 $OOubj))	$
a !233  u{++   (//  
 $O 	XVeVV$+BSVVVWWW;	)) ?I ? ?;$? ? ?  

	;;y)))" 	 	 	
 &44: 4 4 4  	s   -F 
F*F%%F*c                F   | j         s|                     d          gS | j        dk    rb| j        d         dk    rQ|                     t
          j                  }t          d |D                       r|                     d          gS |S | j        }|j        dk    r|d         }t          j
        |d          }d	}||u st          |t                    r|j        |u r| j        }t          || j                  }t!          |          }|                     ||
          gS )zz
        Attempt to coerce any object types to better types. Return a copy
        of the block (if copy = True).
        Fdeepr   r   c              3  6   K   | ]}|j         j        d k    V  dS )ONrd   rz   )r   blks     rg   	<genexpr>z Block.convert.<locals>.<genexpr>  s+      ;;S39>S(;;;;;;ri   r   T)convert_non_numericNr   )rm   copyr]   r   r   rZ   convertallr[   r   maybe_convert_objectsrx   r>   _ndarrayr^   r   r   r   )rf   blocksr[   r   r^   s        rg   r   zBlock.convert  s=    ~ 	+II5I))**9>>djmq00++EM::F;;F;;;;; /		u	--..M;! AYF. $
 
 

 z#677  #v--9D'
DI>>
(44

6677ri   Tnumpy_nullableinfer_objectsconvert_stringconvert_integerconvert_booleanconvert_floatingdtype_backendrN   c           	     h    r j         r                                 }n g}t          g          sd |D             S g }|D ]}	|	j        dk    s j        d         dk    r|	gn t          |	                                          }
fd|
D             }t           fd|D                       r*|                    |		                    d                     t          ||
d	          D ]8\  }}|                    |                    ||j        dk    
                     9|S )Nc                :    g | ]}|                     d           S )Fr   )r   )r   bs     rg   r   z(Block.convert_dtypes.<locals>.<listcomp>  s&    5551AFFF&&555ri   r   r   c                D    g | ]}t          |j                  S r`   )r   r[   )r   r  r  r	  r  r  r
  r  s     rg   r   z(Block.convert_dtypes.<locals>.<listcomp>%  sL         H"##$!!   ri   c              3  .   K   | ]}|j         k    V  d S rk   r   )r   rd   rf   s     rg   r   z'Block.convert_dtypes.<locals>.<genexpr>1  s*      ;;55DJ&;;;;;;ri   Fr   Tstrict)rd   squeeze)rm   r   anyr]   r   listr   r   r   r   zipr   )rf   r  r  r  r  r	  r
  blksr   r   sub_blksdtypesrd   r  s   ```````       rg   r   zBlock.convert_dtypes  s     	T^ 	<<>>DD6DP
 
 	6 655555 	G 	GC Q$*Q-1*<*<$szz||BTBT          "  F ;;;;F;;;;; 

3888//000>>> G Gq

188%18EEFFFFG
ri   rO   c                    | j         j        S rk   )r[   rd   re   s    rg   rd   zBlock.dtype=  s     {  ri   raiserd   errorsrQ   r  c                   | j         }|rF|j        dk    r;t          |          r,|j        d         dk    rt	          d          |dddf         }t          |||          }t          |          }d}t          |j        |j                  r| j	        }| 
                    ||          }|j        | j        k    r:t          d| j        j         d	| j         d
|j        j         d	|j         d	          |S )a  
        Coerce to the new dtype.

        Parameters
        ----------
        dtype : np.dtype or ExtensionDtype
        errors : str, {'raise', 'ignore'}, default 'raise'
            - ``raise`` : allow exceptions to be raised
            - ``ignore`` : suppress exceptions. On error return original object
        squeeze : bool, default False
            squeeze values to ndim=1 if only one column is given

        Returns
        -------
        Block
        r   r   r   z*Can not squeeze with more than one column.N)r  r   zcannot set astype for dtype (z [z]) to different shape (z]))r[   r]   r   r   
ValueErrorr   r   r   rd   r^   r   r   rt   )rf   rd   r  r  r[   r   r^   newbs           rg   r   zBlock.astypeB  s)   .  	"v{a'',?,F,F'|A!## !MNNNAqqqD\F&vuVDDD
(44
&,
(899 	9Dz55:##6JO6 6'+z6 6JO6 6'+z6 6 6  
 ri   nan)na_repquotingr   c               `    t          | j        |||||          }|                     |          S )z"convert to our native types format)r   r!  float_formatdate_formatdecimal)rG   r[   r   )rf   r#  r$  r%  r   r!  r   s          rg   rG   zBlock.get_values_for_csvp  s@    
 $K%#
 
 
 v&&&ri   r   c                   | j         }|r|                                }d}n|                                }| j        } t	          |           || j        | j        |          S )zcopy constructorNr   )r[   r   viewr^   r   r   r]   )rf   r   r[   r^   s       rg   r   z
Block.copy  s_      	[[]]FDD[[]]F9DtDzz&DNQUVVVVri   inplacec                h    |r| j                                         s| S |                     |          S )Nr   )r^   r   r   )rf   r(  r   s      rg   _maybe_copyzBlock._maybe_copy  s8     	492244 	Kyydy###ri   c                ^    d }| }|r#| j                                         rd}n| j         }||fS )NT)r^   r   )rf   r(  r^   r   s       rg   _get_refs_and_copyzBlock._get_refs_and_copy  sD    { 	!y&&(( !yTzri   masknpt.NDArray[np.bool_] | Nonec                   | j         }|                     |          s|                     |d          gS |t          j        ||          }|                                s|                     |d          gS |                     |          s| j        dk    r=t          |          r.|                     |          }t          |j         ||           |gS | j	        dk    s| j
        d         dk    rh|	|t          u r-|                     t          j        t                              }n|                     |d          }|                    ||d|	          S g }t#          |                                           D ]J\  }}	|                    t)          |                               |	||d|||dz            	                     K|S )
z
        replace the to_replace value with value, possible to create new
        blocks here this is just a call to putmask.
        Fr   Nstringr   r   r   T
to_replacer   r(  r-  )r[   r   r*  r0   mask_missingr  rd   r,   r2   r]   r   r   r   ry   rX   r   replacer   r   r   r   )
rf   r3  r   r(  r-  r[   r   r  r   r   s
             rg   r5  zBlock.replace  s   " %%j11 	;
 $$W5$99::<'
;;Dxxzz %	 $$W5$99::##E**  	tzX/E/E%PU,,/E ""7++CCJe4445LY!^^tz!}11}kk"(6"2"23311%1OO;;%	     F"4;;==11 	 	2JJ&&#-# $!!a!e)_ '      Mri   c                "   t          |          s,|                     |          s|                     d          gS t          |          r'| j        t          dfvr|                     d          gS |                     |          sG| j        dk    rt          |          s-|                     t          j        t                              }n|                     |          }t          j	        |          }t          |j        |||           |gS )a  
        Replace elements by the given value.

        Parameters
        ----------
        to_replace : object or pattern
            Scalar to replace or regular expression to match.
        value : object
            Replacement object.
        inplace : bool, default False
            Perform inplace modification.
        mask : array-like of bool, optional
            True indicate corresponding element is ignored.

        Returns
        -------
        List[Block]
        Fr   r0  )r,   r   r   rd   rX   r   ry   r*  recompiler8   r[   )rf   r3  r   r(  r-  r   rxs          rg   _replace_regexzBlock._replace_regex  s   4 Z   	+)?)?
)K)K 	+ II5I))** 	+FH3E!E!E II5I))** ""5))	..2jH.D.Du.DKK 0 011EE$$W--EZ
##elBt444wri   src_listIterable[Any]	dest_listSequence[Any]regexc           	          j          fdt          ||d          D             }|s                     d          gS t          |          dz
  }t	          j                  r!t                     fd|D             }nfd|D             }|rt          |          } g}t          t          ||d                    D ]4\  }	\  \  }
}}g }t          |          D ]\  }}t          |          dk    r|}n&|}t          |t                    rJ |||dz            }|                    |
|||	          }|	|k    rd
 t           j        j                  D             }|D ]v}|j         j        u rHt          |          |v r6| ur2 j        j                            |t          |                              X|j        j                                         w|                    |           |}6|S )z:
        See BlockManager.replace_list docstring.
        c                |    g | ]8\  }}                     |          sj        d k    %t          |          4||f9S )r0  )r   rd   r,   )r   xyrf   s      rg   r   z&Block.replace_list.<locals>.<listcomp>#  s[     
 
 
1&&q))
 /3jH.D.Dq.D F.D.D.Dri   Tr  Fr   r   c           	   3  d   K   | ]*}t          t          |d                              V  +dS )r   )r?  r-  N)r1   r7   )r   r   na_maskr?  r[   s     rg   r   z%Block.replace_list.<locals>.<genexpr>1  s[       6 6  #+FAaDGTTT 6 6 6 6 6 6ri   c              3  N   K   | ]}t          j        |d                    V   dS )r   N)r0   r4  )r   r   r[   s     rg   r   z%Block.replace_list.<locals>.<genexpr>9  s4      GGAW)&!A$77GGGGGGri   )r3  r   r-  r(  r?  c                D    i | ]\  }}t           |                      |S r`   )id)r   r   r  s      rg   
<dictcomp>z&Block.replace_list.<locals>.<dictcomp>`  s7     $ $ $'+q!1133$ $ $ri   )r[   r  r   r   r"   rd   r.   r  r   rx   rb   _replace_coercer^   referenced_blocksrH  popclearr   )rf   r;  r=  r(  r?  pairssrc_lenmasksrbr   srcdestr-  new_rbblk_numr   mmibr   self_blk_idsr  rE  r[   s   `   `                @@rg   replace_listzBlock.replace_list  s    
 
 
 
Hi===
 
 

  	+II5I))**e**q.6<(( 	H F||mG6 6 6 6 6 6 	6 6 6EE HGGGGGGE  	 KKE V&/E50N0N0N&O&O +	 +	"A"dT"$F
 !*" $& $&r77a<<AAC)#t44444Ggk12A
 ,,"# -   <<$ $/89T/U/U$ $ $L $ 	= 	=6TY..!!uu44$ $	 ; ? ?RPQUU@S T T T F4::<<<<f%%%%BB	ri   npt.NDArray[np.bool_]c                4   t          ||          r|                     ||||          S ||                                r| j                                        }|                     t          j        t                              }|s|	                    d          }n3|r1|r/|j                                        r|	                    d          }t          |j        ||           |gS | 	                    d          gS |                     ||||          S )ai  
        Replace value corresponding to the given boolean array with another
        value.

        Parameters
        ----------
        to_replace : object or pattern
            Scalar to replace or regular expression to match.
        value : object
            Replacement object.
        mask : np.ndarray[bool]
            True indicate corresponding element is ignored.
        inplace : bool, default True
            Perform inplace modification.
        regex : bool, default False
            If true, perform regular expression substitution.

        Returns
        -------
        List[Block]
        )r(  r-  NTr   Fr2  )r9   r:  r  r^   r   r   ry   rd   rX   r   r2   r[   r5  )rf   r3  r   r-  r(  r?  has_refr   s           rg   rJ  zBlock._replace_coerceq  s1   < E:.. 	&&	 '    }88:: 	 "i5577GRXf%5%566B" 0WW$W//  0W 01F1F1H1H 0WW$W//#BItU;;;4K		u	--..<<%	     ri   arg
np.ndarrayc                    |S zA
        For compatibility with 1D-only ExtensionArrays.
        r`   rf   r]  s     rg   _maybe_squeeze_argzBlock._maybe_squeeze_arg  s	     
ri   c                    |S r`  r`   )rf   r   s     rg   _unwrap_setitem_indexerzBlock._unwrap_setitem_indexer  s	     ri   rT   c                    | j         j        S rk   )r[   r   re   s    rg   r   zBlock.shape  s    {  ri   r   )int | tuple[int, int] | tuple[slice, int]c                    | j         |         S rk   r[   )rf   r   s     rg   igetz
Block.iget  s     {1~ri   4slice | npt.NDArray[np.bool_] | npt.NDArray[np.intp]c                    | j         |         S )zreturn a slice of my valuesrh  )rf   r   s     rg   r   zBlock._slice  s    
 {6""ri   r   c                Z    |r| j                                         | _         || j         |<   dS )a  
        Modify block values in-place with new item value.

        If copy=True, first copy the underlying values in place before modifying
        (for Copy-on-Write).

        Notes
        -----
        `set_inplace` never creates a new array or new Block, whereas `setitem`
        _may_ create a new array and always creates a new Block.

        Caller is responsible for checking values.dtype == self.dtype.
        Nr[   r   rf   locsr[   r   s       rg   set_inplacezBlock.set_inplace  s3      	-+**,,DK"Dri   r   axisrM   c                \   | j         }|t          j        u r
| j        }d}nd}t	          j        |||||          }t          | t                    r| j        dk    r|J |dk    r|J || j	        }|j
        | j
        k    r|                     ||          S |                     ||          S )zN
        Take values according to indexer and return them as a block.
        FT)rq  
allow_fillr   r   Nr   )r[   r   
no_defaultr   algostake_ndrx   ExtensionBlockr]   r   rd   r   r   )rf   r   rq  r   r   r[   rs  r   s           rg   rv  zBlock.take_nd  s     ''JJJJ ]G$:*
 
 

 dN++ 	A 	Q<+?+?@AII,"6"67>Ltz))??:|<<<--j,GGGri   new_placementneeds_maskingc                    |                     | j        j        |          \  }}|                    d          }|j        |         }||         }t	          |          }t          ||          g}||fS )a  
        Return a list of unstacked blocks of self

        Parameters
        ----------
        unstacker : reshape._Unstacker
        fill_value : int
            Only used in ExtensionBlock._unstack
        new_placement : np.ndarray[np.intp]
        allow_fill : bool
        needs_masking : np.ndarray[bool]

        Returns
        -------
        blocks : list of Block
            New blocks of unstacked values.
        mask : array-like of bool
            The mask of columns of `blocks` we should keep.
        r   r   r   )get_new_valuesr[   Tr  r   new_block_2d)	rf   	unstackerr   rx  ry  r   r-  r   r  s	            rg   _unstackzBlock._unstack  s    4 %33KMj 4 
 

D xx{{  \$'
%d+M**zR8889t|ri   c                   |                      |          }t          t          j        | j                  }| j        dk    r|j        }t          |||           | j        t          k    rt          |d          }	 t          |j        |          }| j        t          k    r:||         }t          j        |          rt          |t          |          |          }|                     d          } t          t          j        | j        j                  }t#          |t          j                  r&|j        dk    rt          |          dk    r|d         }	 |||<   nv# t$          t&          f$ r%}t          |          rt'          d          | d}~ww xY w# t(          $ r0 |                     |d	          }|                    ||          cY S w xY w| S )
a  
        Attempt self.values[indexer] = value, possibly creating a new array.

        Parameters
        ----------
        indexer : tuple, list-like, array-like, slice, int
            The subset of self.values to set
        value : object
            The value being set

        Returns
        -------
        Block

        Notes
        -----
        `indexer` is a direct slice/positional indexer. `value` must
        be a compatible shape.
        r   Tr   r(  r   )r   .z)setting an array element with a sequence.Nr1  )r   r   ry   r   r[   r]   r~  rF   rd   rl   rE   r   r   r    r4   r   r*  rx   r   r  r   r   setitem)rf   r   r   r[   castedvir   r   s           rg   r  zBlock.setitem9  s   * ,,U33bj$+..9>>XF 	guf555:##!%t<<<E	(u==F zZ''G_#B'' R 9R&QQF##D#11D"*dkm44F&"*-- (&+2B2Bs6{{VWGWGW"(wz*   '' $C  ) ! 	. 	. 	.,,UD,IIB::gu-----	.4 s*   :F E F+ FF7GGc                   |}t          t          j        | j                  }t	          |j        |          \  }}t          |t          t          t          f          rJ |t          j        u r| j        }|                     |          }t          |d          }|r|                     d          gS 	 t!          |j        |          }|                     d          } t          t          j        | j                  }t'          |j        ||           | gS # t(          $ r, | j        dk    s| j        d         dk    rwt/          |          s-|                     |d                              ||          cY S |                                d         }|                     |||                   }|gcY S t          |t          j                  }	g }
t9          |                                           D ]V\  }}|}|	r|d	d	||dz   f         }|d	d	||dz   f         }|                    ||          }|
                    |           W|
cY S w xY w)
aW  
        putmask the data to the block; it is possible that we may create a
        new dtype of block

        Return the resulting block(s).

        Parameters
        ----------
        mask : np.ndarray[bool], SparseArray[bool], or BooleanArray
        new : an ndarray/object

        Returns
        -------
        List[Block]
        Tr   Fr   r  r   r   r1  N)r   ry   r   r[   r5   r~  rx   r)   r+   r(   r   rt  r   r   rE   r   r   rd   r*  r3   r   r]   r   r    r   putmasknonzeror  r   r   r   )rf   r-  new	orig_maskr[   noopr  r   r   is_arrayr   r   nsubmaskr   s                  rg   r  zBlock.putmaskx  sk     	bj$+..%fh55
dcHi#FGGGGG#.  /C**3//Ct444 	+II5I))**#	"(s;;F##D#11D"*dk22F"68T6:::6M  	" 	" 	"yA~~A!!3!3 $C((  66T 7  gdC(() ) ) #llnnQ/Ggs7|<<B4KKK &c2:66
&t{{}}55 + +EArA .1q1u9-'1q1u95G**Wa00C%%c****!!!!7	"s!   3A"D A!I99I4BIIc                   |j         | j         k    sJ t          |t          t          t          f          rJ | j         dk    }t          |          }t          t          j        | j	                  }|}|r|j
        }t          ||           \  }}|r|                     d          gS |t          j        u r| j        }|                     |          }	 t#          |j        |          }|}t'          ||                                |          }	|	|urt+          |          rFt-          |          t-          |          k     r&t          j        | ||           t1          d          |                                }
t          j        |
||	           nt+          |          rdt          |t          j                  sJt-          |          | j        d         k    r,t          j        |                              |j                  }t;          j        | ||          }
n# t<          t>          t@          f$ r | j         dk    s| j        d         dk    r/| !                    |d          }|                    ||          cY S t          |t          j        tD          f          }g }tG          | $                                          D ]V\  }}|}|r|d	d	||dz   f         }|d	d	||dz   f         }|                    ||          }|%                    |           W|cY S w xY w|r|
j
        }
| &                    |
          gS )
a  
        evaluate the block; return result block(s) from the result

        Parameters
        ----------
        other : an ndarray/object
        cond : np.ndarray[bool], SparseArray[bool], or BooleanArray

        Returns
        -------
        List[Block]
        r   Fr   zThis should not be reached; call to np.where above is expected to raise ValueError. Please report a bug at github.com/pandas-dev/pandasr   r   r   r1  N)'r]   rx   r)   r+   r(   r1   r   ry   r   r[   r~  r5   r   r   rt  r   r   r   rd   r4   sumr    r   whereNotImplementedErrorr  r   arrayr   rC   r  r   r   r   r<   r   r   r   r   )rf   r   cond	transposer[   
orig_othericondr  r  altr   r   r  r   r   r   othr  r   s                      rg   r  zBlock.where  s^    yDI%%%%eh	<%HIIIIIIN	!$'' bj$+..
 	XF&vu55t 	+II5I))**CN""OE,,U33;	B )u==F4 E-feiikk5IIC%&& 3u::F+C+CHeVVU333-7  
  
65#....
 !''B&ubj99B E

djn44
 HUOO33FLAAE
 %*E665AAm I'89 	" 	" 	" yA~~A!!3!3 33E53QQ{{:t44444 &ebj.-IJJ
&t{{}}55 + +EArC 2#AAAq1q5yL1"111a!a%i<0G((300C%%c****!!!!-	"t  	XF''((s   H A L  BL L limit
int | Nonec                   t          |d          }| j        sd}n,t          | j                  }t	          | j        |          \  }}|r|                     d          gS |)d||                    | j        j        dz
            |k    <   |r|                     |j	        |          }n| 
                    ||j	                   }t          |          S )z
        fillna on the block with the value. If we fail, then convert to
        block to hold objects instead and try again
        r(  TFr   Nr   )r   r{   r.   r[   r5   r   cumsumr]   r  r~  r  extend_blocks)rf   r   r  r(  r  r-  r   s          rg   fillnazBlock.fillna"  s     &gy99  	=DD$$D)$+t<<JD$ 	+II5I))**>CDT[-122U:; 	-,,tvu--CC**UTVG,,CS!!!ri   r(  r  
limit_areamethodrP   r  #Literal['inside', 'outside'] | Nonec               :   | j         s|                     d          gS |                     |          \  }}t          t          | j                  }|j                            ||||          j        }t          |d          }	| 	                    |	|          gS )NFr   )r  r  r  r   Tr   r   )
r{   r   r,  r   r>   array_valuesr~  _pad_or_backfillrE   r   )
rf   r  r(  r  r  r   r^   r   r   datas
             rg   pad_or_backfillzBlock.pad_or_backfillC  s       	+II5I))**,,W55
d '):;;V,,!	 - 
 

  	 Zt<<<**4d*;;<<ri   forward)r(  r  limit_directionr  rR   indexrV   r  &Literal['forward', 'backward', 'both']c                  t          |d          }|dk    rt          j        |           | j        s|                     d          gS | j        t          k    r"ddd| j                 }t          | d          | 	                    |          \  }	}
 | j
        j        d|| j        d	z
  |||||	d
|}t          |d          }|                     ||
          gS )Nr(  asfreqFr   Series	DataFrame)r   r   z& cannot interpolate with object dtype.r   )r  rq  r  r  r  r  r   Tr   r   r`   )r   r0   clean_fill_methodr{   r   rd   rl   r]   r   r,  r  interpolaterE   r   )rf   r  r  r(  r  r  r  r   rt   r   r^   r   r  s                rg   r  zBlock.interpolate^  s    &gy99X%f---  	+II5I))**:##K00;DtKKKLLL,,W55
d 3T&2 	
Q+!	
 	
 	
 	

 Zt<<<**4d*;;<<ri   r  c                |    t          j        | j        j        |d          j        }|                     |          gS )z'return block for the diff of the valuesr   rq  rh  )ru  diffr[   r~  r   )rf   r  r   s      rg   r  z
Block.diff  s8    
 Zqq999;
z2233ri   periodsr   c                r   | j         dz
  }t          j        |          s| j        t          k    rt          d          |                     |          }	 t          | j        |          }t          t          j
        | j                  }t          ||||          }|                     |          gS # t          $ r| | j        j        dvst!          || j                  s(t#          j        dt&          t)                                 |                     |d          }|                    ||          cY S w xY w)	z+shift the block by periods, possibly upcastr   zfill_value must be a scalarrw   zshifting with a fill value that cannot be held by original dtype is deprecated and will raise in a future version. Explicitly cast to the desired dtype before shifting instead.
stacklevelFr1  r{  )r]   r   r!   rd   rl   r  r   r   r   ry   r   r[   r:   r   r   rz   r-   warningswarnr   r   r   )rf   r  r   rq  r  r[   r   r   s           rg   r:   zBlock.shift  sW    y1} }Z(( 	<TZ:-E-E :;;;11*==
	< )
 F* "*dk22Fvwf==J..z::;;' ! 	< 	< 	<ze++3HDJ4 4+ ( #/11    ,,Z,OOB88G
8;;;;;	<s   B0 0BD65D6linearqsinterpolationrS   c                    | j         dk    sJ t          |          sJ t          | j        t	          j        |j                  |          }t          |d          }t          || j	                  S )a  
        compute the quantiles of the

        Parameters
        ----------
        qs : Index
            The quantiles to be computed in float64.
        interpolation : str, default 'linear'
            Type of interpolation.

        Returns
        -------
        Block
        r   r   r|  )
r]   r    r6   r[   ry   asarray_valuesr   r  r   )rf   r  r  r   s       rg   quantilezBlock.quantile  so    * yA~~~~B bj.D.DmTT $F333Fdn====ri   decimalsc                n   | j         r| j        ret          | j        t          t
          t          f          r(t          j        dt          t                                 |                     d          S | j                            |          }d}|| j        u r| j        }|                     ||          S )ax  
        Rounds the values.
        If the block is not of an integer or float dtype, nothing happens.
        This is consistent with DataFrame.round behavior.
        (Note: Series.round would raise)

        Parameters
        ----------
        decimals: int,
            Number of decimal places to round to.
            Caller is responsible for validating this
        zbobj.round has no effect with datetime, timedelta, or period dtypes. Use obj.dt.round(...) instead.r  Fr   Nr   )
is_numericr~   rx   r[   r;   r@   r?   r  r  UserWarningr   r   roundr^   r   )rf   r  r[   r^   s       rg   r  zBlock.round  s      
	)$, 
	)$+~{'STT  G/11	    99%9(((
 ""8,,T[  9D))&t)<<<ri   c                   t          |          s|g}| j        dk    rot          t          j        | j                  }t          j        ||          }| j                            |          } t          |           ||| j                  gS t          j	        |          | j        j
        d         k    rt          t          j        || j        j
        d         gg          }| j        j        }g }d}| j                                        r| j        nd}|D ]t}||dz   k    rnf| j        |dz   |ddf         }||dz   |         }	 t          |           |t!          |	          | j        |          }
|                    |
           |}u|S )a  Deletes the locs from the block.

        We split the block to avoid copying the underlying data. We create new
        blocks for every connected segment of the initial block that is not deleted.
        The new blocks point to the initial array.
        r   r   r]   r   r   Nr   )r    r]   r   ry   r   r[   deleter   r   maxr   
IndexErrorconcatenateas_arrayr^   r   r   r   )rf   r   r[   r   mgr_locs_arr
new_blocksprevious_locr^   idxro  r   s              rg   r  zBlock.delete  s    C   	%C9>>"*dk22FYvs++F~,,S11HDJJv	JJJKK6#;;$++A... ncDK$5a$8#9:;;~."$
 !I3355?tyy4 	 	ClQ&&& \A%5%;QQQ%>?#L1$4s$:;T$ZZnT&:&:QU   !!"%%%LLri   c                     t          |           )(return a boolean if I am possibly a viewr   re   s    rg   is_viewzBlock.is_view5  s     "$'''ri   r<   c                     t          |           )zP
        The array that Series.array returns. Always an ExtensionArray.
        r  re   s    rg   r  zBlock.array_values:      
 "$'''ri   DtypeObj | Nonec                     t          |           )z
        return an internal format, currently just the ndarray
        this is often overridden to handle to_dense like operations
        r  r|   s     rg   
get_valueszBlock.get_valuesA  r  ri   ra   rb   )ra   r   )r   r   ra   r   )NN)r   r   r^   r   ra   rZ   )r   r   r^   r   ra   r   )ra   r   )ra   r\   )r   r   ra   r   )r   r   ra   r   F)r   r   r   r   r   rb   ra   r   )r   r   ra   rb   )r   rL   ra   rb   ra   r   )ra   rZ   )r   rL   ra   r   )ra   r   )r   rb   ra   rZ   )TTTTTr  )r  rb   r  rb   r  rb   r  rb   r	  rb   r
  rN   ra   r   )ra   rO   )r  F)rd   rO   r  rQ   r  rb   ra   rZ   )r   r   ra   rZ   )r   rb   ra   r   )T)r(  rb   r   rb   ra   r   )r(  rb   )FN)r(  rb   r-  r.  ra   r   )r(  rb   ra   r   )FF)
r;  r<  r=  r>  r(  rb   r?  rb   ra   r   )TF)r-  rZ  r(  rb   r?  rb   ra   r   )r]  r^  ra   r^  ra   rT   )r   rf  ra   r^  )r   rj  ra   rL   r[   rL   r   rb   ra   r   )r   r   rq  rM   r   r   ra   rZ   rx  r   ry  rZ  NFr  r  r(  rb   ra   r   
r  rP   r(  rb   r  r  r  r  ra   r   )r  rR   r  rV   r(  rb   r  r  r  r  r  r  ra   r   )r  r\   ra   r   rk   r  r\   r   r   ra   r   )r  )r  rV   r  rS   ra   rZ   )r  r\   ra   r   ra   r<   rd   r  ra   r^  )Ir   
__module____qualname____doc____annotations__	__slots__r  r	   r   rh   rm   rp   rr   ru   r{   propertyr~   r   r   r   r   setterr   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   rd   r   rG   r   r*  r,  r5  r:  rY  rJ  rb  rd  r   ri  r   rp  r   rt  rv  r  r  r  r  r  r  r  r  r:   r  r  r  r  r  r  r`   ri   rg   rZ   rZ      s          ('''IIIIJ
3 3 3 ^ U3 / / / ^ U/ 6 6 6 ^ U6 % % % ^ U% 6 6 ^ U6 " " " ^ U" 3 3 3 X U3 , , U, < < ^ U<   U    X _& & & _&  ,0'+	Q Q Q Q UQ"  ,0'+	R R R R UR 	 	 	 U	       U  J J J UJ J J J UJ RW
J 
J 
J 
J U
J 6 6 6 U6
 ) ) ) U)" - - - U- + + + U+    U*    U    U2 7 7 7 U7r "8 "8 "8 U"8L ## $ $!%&6, , , , ,b ! ! ! ^ U!  &	+ + + + U+Z CHRV' ' ' ' ' U' 
W 
W 
W U
W$ $ $ $ $
    U 
 -1@ @ @ @ U@D 
 - - - - U-^ 
 Y Y Y Y UYv  7 7 7 7 U7z      ! ! ! X!   # # # ## # # # #$ 
 /3>%H %H %H %H U%HN* * * *\= = = =~A" A" A" A"Fe) e) e) e)T !	" " " " "J  :>= = = = = =6   BK:>(= (= (= (= (= U(=T 4 4 4 U4,< ,< ,< ,< ,<\  08> > > > U>: "= "= "= U"=N. . . .` ( ( ( X( ( ( ( X(( ( ( ( ( ( (ri   rZ   c                      e Zd ZU dZded<   ed d!d            Zed             Zed"d            Zed"d            Z	ed" fd            Z
eed#d                        Zed d$d            Zeddddd%d            Z xZS )&EABackedBlockz>
    Mixin for Block subclasses backed by ExtensionArray.
    r<   r[   Nr  r\   r   r   ra   r   c                z    | j         j                            ||          j        }|                     |          gS )z
        Shift the block by `periods`.

        Dispatches to underlying ExtensionArray and re-boxes in an
        ExtensionBlock.
        )r  r   )r[   r~  r:   r   )rf   r  r   r   s       rg   r:   zEABackedBlock.shiftP  s:     []((Z(PPR
**:6677ri   c                   |}|}|                      |          }|                     |          }| j        }|j        dk    r|j        }t          |||           	 |||<   | S # t          t          f$ r t          | j	        t                    r/|                     |d          }|                    ||          cY S t          | t                    r/|                     |d          }|                    ||          cY S  w xY w)a7  
        Attempt self.values[indexer] = value, possibly creating a new array.

        This differs from Block.setitem by not allowing setitem to change
        the dtype of the Block.

        Parameters
        ----------
        indexer : tuple, list-like, array-like, slice, int
            The subset of self.values to set
        value : object
            The value being set

        Returns
        -------
        Block

        Notes
        -----
        `indexer` is a direct slice/positional indexer. `value` must
        be a compatible shape.
        r   Tr1  )rd  rb  r[   r]   r~  rF   r  r   rx   rd   r%   r   r  NDArrayBackedExtensionBlock)rf   r   r   orig_indexer
orig_valuer[   r   s          rg   r  zEABackedBlock.setitem]  s*   0 
..w77''..;! XFguf555	#F7O K I& 	 	 	$*m44 
00T0RRzz,
;;;;;D"=>> 00T0RRzz,
;;;;; 	s   A! !AD;AD?Dc                   | j         j        }t          |          }|}|}|                     |          }|                     |          }|t          j        u r| j        }t          ||           \  }}|r|                     d          gS 	 |	                    ||          j        }n# t          $ r  t          t          f$ r | j        dk    s| j        d         dk    rt          | j        t"          t$          f          r|                     |d          }	| j        dk    rEt          |t(          j                  r+|j        dk    r t-          |	j                  s|d d d f         }|	                    ||          cY S t          | t0                    r/|                     |d          }	|	                    ||          cY S  t          |t(          j        t2          f          }
g }t5          |                                           D ]V\  }}|}|
r|d d ||dz   f         }|d d ||dz   f         }|                    ||          }|                    |           W|cY S w xY w|                     |          }|gS )NFr   r   r   r1  r   )r[   r~  r1   rb  r   rt  r   r5   r   _wherer   r  r   r]   r   rx   rd   r%   rA   r   ry   r   r   r  r  r<   r   r   r   r   )rf   r   r  arrr  	orig_condr  r  r   r   r  r   r   r   r  r  r   s                    rg   r  zEABackedBlock.where  s   km!$''
	''..&&t,,CN""OE&sTE22t 	+ II5I))**)	"D%002JJ" 	 	 	I& %	" %	" %	"yA~~A!!3!3dj=+*FGG 55jRW5XXC	Q&y"*== '%Na// 3CI > > 0 %.aaag$6	99Z;;;;;&ABB  55jRW5XXC99Z;;;;;  &j2:~2NOO
&t{{}}55 + +EAr"A 5&qqq!a!e)|4'1q1u95G((1g..C%%c****!!!!K%	"N ''
33ts!   B* *CI'AI'BI'&I'c                d   t          |          }|t          j        u r| j        }|}|}|                     |          }|                     |          }|                                s|                     d          gS |                     d          } | j        }|j	        dk    r|j
        }	 |                    ||           ni# t          $ r  t          t          f$ rJ | j	        dk    s| j        d         dk    rt!          | j        t$                    r/|                     |d          }|                    ||          cY S t!          | t*                    r/|                     |d          }|                    ||          cY S  t!          |t,          j        t0          f          }g }t3          |                                           D ]V\  }	}
|}|r|d	d	|	|	dz   f         }|d	d	|	|	dz   f         }|
                    ||          }|                    |           W|cY S w xY w| gS )
z+
        See Block.putmask.__doc__
        Fr   Tr  r   r   r   r1  N)r1   r   rt  r   rb  r  r   r*  r[   r]   r~  _putmaskr   r   r  r   rx   rd   r%   r   r  r  ry   r   r<   r   r   r   )rf   r-  r  orig_newr  r[   r   r  r   r   r   r  r  r   s                 rg   r  zEABackedBlock.putmask  sp   
 "$''#.  /C	%%c**&&t,,xxzz 	+II5I))**--;!XF$	"OOD#&&&&" 	 	 	:& 	" 	" 	"yA~~A!!3!3dj-88  55hPT5UUC;;y(;;;;;&ABB  55hPT5UUC;;y(;;;;;  &h^0LMM
&t{{}}55 + +EAr A 3$QQQAE	\2'1q1u95G**Wa00C%%c****!!!!?	"B vs!   .C A?H,AH,BH,+H,c                *   | j         dk    rU| j                            |          }| j                            |          } t	          |           ||| j                   gS | j        j         dk    rg S t                                          |          S )Nr   r  )r]   r[   r  r   r   super)rf   r   r[   r   	__class__s       rg   r  zEABackedBlock.delete  s     9>>['',,F~,,S11HDJJv	JJJKK[""Iww~~c"""ri   c                    | j         S rk   rh  re   s    rg   r  zEABackedBlock.array_values   s     {ri   rd   r  r^  c                    | j         }|t          k    r|                    t                    }t	          j        |                              | j                  S )zS
        return object dtype as boxed values, such as Timestamps/Timedelta
        )r[   rl   r   rX   ry   r  r   r   )rf   rd   r[   s      rg   r  zEABackedBlock.get_values%  sG    
 !KJ]]6**Fz&!!))$*555ri   Fr  r  rP   r(  rb   r  r  r  r  c               >   | j         }||d}dt          j        |j                  j        v r||d<   n&|$t          t          |          j         d          |j        dk    r |j	        j        di |j	        }n |j        di |}| 
                    |          gS )N)r  r  r  z does not implement limit_area (added in pandas 2.2). 3rd-party ExtensionArray authors need to add this argument to _pad_or_backfill.r   r`   )r[   inspect	signaturer  
parametersr  r   r   r]   r~  r   )rf   r  r(  r  r  r[   r   r   s           rg   r  zEABackedBlock.pad_or_backfill0  s     ,2U!C!C7,V-DEEPPP#-F<  #%<<( A A A   ;!22<<V<<>JJ00::6::J**:6677ri   rk   r  r  r  r  r  )r   r  r  r  r  r	   r:   r  r  r  r  r   r  r  r  __classcell__r   s   @rg   r  r  I  sh          

8 
8 
8 
8 U
8 4 4 U4l ? ? ? U?B ; ; ; U;z 	# 	# 	# 	# 	# U	#    ^ U 6 6 6 6 U6 
  :>8 8 8 8 8 U8 8 8 8 8ri   r  c                       e Zd ZU dZded<   	 	 d%d& fdZed'd            Zd(dZd)d*dZ	d Z
d Zed+d            Zed+d            Zd,dZed-d            Zd.d$Z xZS )/rw  z
    Block for holding extension types.

    Notes
    -----
    This holds all 3rd-party extension array types. It's also the immediate
    parent class for our internal extension types' blocks.

    ExtensionArrays are limited to 1-D.
    r<   r[   NFr  r  r(  rb   ra   r   c                   t          | j        t          t          f          rOt          | j        t                    r|t	          d          t                                          |||          S | j        r| j        j	        s| j
        }| j        }n|                     |          \  }}	 | j                            |||          }nq# t          $ rd d }| j                            ||          }t          j        dt          | j                  j         dt"          t%                                 Y nw xY w|                     ||          gS )	Nzlimit must be None)r   r  r(  )r   r  r   )r   r  zExtensionArray.fillna added a 'copy' keyword in pandas 2.1.0. In a future version, ExtensionArray subclasses will need to implement this keyword or an exception will be raised. In the interim, the keyword is ignored by .r  r   )rx   rd   r%   rA   r  r  r  r{   r[   _hasnar^   r,  r   r  r  r   r   r   r   r   )rf   r   r  r(  r^   r   r   r   s          rg   r  zExtensionBlock.fillna[  s    dj=+">?? 	$*m44 79J !566677>> "   
  	T[%7 	9DJJ0099JD$![//e5t/TT

   ![//e5/II
 5 DK((1	5 5 5
 #/11	 	 	 	 	 	" **:D*AABBs   -C A+D98D9rT   c                    | j         dk    rt          | j                  fS t          | j                  t          | j                  fS )Nr   )r]   r   r[   r   re   s    rg   r   zExtensionBlock.shape  sA     9>>$$&&4>""C$4$444ri   r   rf  c                n   t          |t                    r|\  }}t          j        |          s|dk    rt	          |  d          t          |t
                    r/|dk     r|t          | j                  z  }| j        ||dz            S | j        |         S |dk    rt	          |  d          | j        S )Nr   z only contains one itemr   )rx   tuplecomis_null_slicer  r   r   r[   )rf   r   colr   s       rg   ri  zExtensionBlock.iget  s     a 	HC$S)) CcQhh D!A!A!ABBB#u%% 2 773t{+++C {3q=11;s##Avv D!A!A!ABBB;ri   rL   r   r   c                ^    |r| j                                         | _         || j         d d <   d S rk   rm  rn  s       rg   rp  zExtensionBlock.set_inplace  s4      	-+**,,DKAAAri   c                @   t          |t          j        t          f          r8|j        | j        j        dz   k    r |j        d         dk    sJ |dddf         }nDt          |t                    r/|j        d         dk    sJ |                    dd          j	        }|S )z@
        If necessary, squeeze a (N, 1) ndarray to (N,)
        r   Nr   r  )
rx   ry   r   r<   r]   r[   r   r(   _ixsr  ra  s     rg   rb  z!ExtensionBlock._maybe_squeeze_arg  s     sRZ899	.DK,q000 9Q<1$$$$ aaad)CC\** 	. 9Q<1$$$$((11(%%-C
ri   c                V   t          |t                    rt          |          dk    rt          d |D                       rU|\  }}|j        dk    r)|dk                                    r|j        d         dk    st          d          |dddf         }nt          j        |d                   r|d         dk    r	|d         }nbt          j
        |d                   r	|d         }n?t          |d                   r|d         d         dk    r	|d         }nt          d          |S )zw
        Adapt a 2D-indexer to our 1D values.

        This is intended for 'setitem', not 'iget' or '_slice'.
        r   c              3  `   K   | ])}t          |t          j                  o
|j        d k    V  *dS )r   N)rx   ry   r   r]   )r   rB  s     rg   r   z9ExtensionBlock._unwrap_setitem_indexer.<locals>.<genexpr>  s9      NN:a,,<1NNNNNNri   r   r   zPThis should not be reached. Please report a bug at github.com/pandas-dev/pandas/N)rx   r  r   r   sizer   r  r   
is_integerr  r  r    )rf   r   firstseconds       rg   rd  z&ExtensionBlock._unwrap_setitem_indexer  sO    gu%% 	#g,,!*;*;
 NNgNNNNN  'vK1$$&A+):):)<)<$QSTATAT-8    1+
++ 
a!!*"71:.. 
!!*gaj)) gajmq.@.@!!* *4   ri   c                    dS )z,Extension arrays are never treated as views.Fr`   re   s    rg   r  zExtensionBlock.is_view  s	     uri   c                $    | j         j        j        S rk   )r[   rd   _is_numericre   s    rg   r  zExtensionBlock.is_numeric  s    { ,,ri   r   rj  c                   | j         dk    rht          |t                    st          d|          t	          d          |         }t          |          st          d|          t          d          }| j        |         S )z
        Return a slice of my values.

        Parameters
        ----------
        slicer : slice, ndarray[int], or ndarray[bool]
            Valid (non-reducing) indexer for self.values.

        Returns
        -------
        ExtensionArray
        r   z+invalid slicing for a 1-ndim ExtensionArrayr   N)r]   rx   r   r   ranger   r[   )rf   r   new_locss      rg   r   zExtensionBlock._slice  s    & 9>> fe,, $A6   Qxx'Hx== $A6   4[[F{6""ri   r   r   c                r    | j         |         } t          |           || j        | j        | j                  S )zN
        Perform __getitem__-like specialized to slicing along index.
        )r]   r^   )r[   r   r   r]   r^   )rf   r   r   s      rg   slice_block_rowszExtensionBlock.slice_block_rows!  s5     [(
tDzz*dn4949UUUUri   rx  r   ry  rZ  c                     |j         \  }}|j        |         }||         } fdt          t          ||d                    D             }||fS )Nc           	         g | ]Q\  }\  }} t                    j                            ||                    t          |          d          RS ))rs  r   r   r   )r   r[   taker   )r   r   r   placer   ry  rf   s       rg   r   z+ExtensionBlock._unstack.<locals>.<listcomp>G  s     
 
 
 $#GU DJJ  a(8Z !   u%%  
 
 
ri   Tr  )arange_resultr~  r   r  )rf   r  r   rx  ry  r   r-  r  s   ` ` `   rg   r  zExtensionBlock._unstack+  s     %2
D  \$'
%d+
 
 
 
 
 
 (1Jd;;;( (
 
 
 t|ri   r  r  r  )r   rf  r  r  r  )r   rj  ra   r<   )r   r   ra   r   r  )r   r  r  r  r  r  r   r   ri  rp  rb  rd  r  r  r  r   r	   r#  r  r  r  s   @rg   rw  rw  M  se        	 	 
 !	(C (C (C (C (C (C (CT 5 5 5 ^5   8           ,( ( (T    X
 - - - ^-&# &# &# &#P V V V UV) ) ) ) ) ) ) )ri   rw  c                  p    e Zd ZU ded<   dZedd            Zedd            ZdddZe	dd            Z
d	S )
NumpyBlockr^  r[   r`   ra   rb   c                    | j         j        duS r  N)r[   basere   s    rg   r  zNumpyBlock.is_view[  s     {t++ri   r<   c                *    t          | j                  S rk   )r>   r[   re   s    rg   r  zNumpyBlock.array_values`  s    "4;///ri   Nrd   r  c                d    |t           k    r| j                            t                     S | j        S rk   )rl   r[   r   r|   s     rg   r  zNumpyBlock.get_valuesd  s+    J;%%j111{ri   c                0    | j         j        }|j        }|dv S )Nfciub)r[   rd   rz   )rf   rd   rz   s      rg   r  zNumpyBlock.is_numerici  s    !zwri   r  r  rk   r  )r   r  r  r  r  r  r  r  r  r   r  r`   ri   rg   r*  r*  W  s         I, , , X, 0 0 0 X0    
    ^  ri   r*  c                  6    e Zd ZU dZded<   edd            ZdS )	r  z8
    Block backed by an NDArrayBackedExtensionArray
    rW   r[   ra   rb   c                (    | j         j        j        duS r,  )r[   r  r-  re   s    rg   r  z#NDArrayBackedExtensionBlock.is_viewx  s     {#(44ri   Nr  )r   r  r  r  r  r  r  r`   ri   rg   r  r  q  sL           ('''5 5 5 X5 5 5ri   r  c                  &    e Zd ZU dZdZdZded<   dS )DatetimeLikeBlockz*Block for datetime64[ns], timedelta64[ns].r`   FzDatetimeArray | TimedeltaArrayr[   N)r   r  r  r  r  r  r  r`   ri   rg   r5  r5    s.         44IJ******ri   r5  r[   rL   ra   c                <   t          | t          j                  rIt          |           } t	          | j        j        t                    rt          j        | t                    } t          | t          t          f          r| j        |                     d          } | S )a:  
    Input validation for values passed to __init__. Ensure that
    any datetime64/timedelta64 dtypes are in nanoseconds.  Ensure
    that we do not have string dtypes.

    Parameters
    ----------
    values : np.ndarray or ExtensionArray

    Returns
    -------
    values : np.ndarray or ExtensionArray
    r   N)rx   ry   r   rD   
issubclassrd   r   r   r  rX   r;   r@   freq
_with_freqrh  s    rg   r   r     s      &"*%% 4/77fl'-- 	4XfF333F&=.9:: )v{?V""4((Mri   rd   rO   type[Block]c                    t          | t                    rt          S t          | t                    rt          S t          | t
                    rt          S | j        }|dv rt          S t          S )z
    Find the appropriate Block subclass to use for the given values and dtype.

    Parameters
    ----------
    dtype : numpy or pandas dtype

    Returns
    -------
    cls : class, subclass of Block
    Mm)	rx   r#   r5  r'   r  r$   rw  rz   r*  r   s     rg   get_block_typer=    sj     %))   	E;	'	' **	E>	*	*  :Dt||  ri   r   r   r^   r   c                f    t          | j                  }t          |           }  || d||          S )Nr   r]   r   r^   )r=  rd   r   )r[   r   r^   klasss       rg   r  r    s9     6<((E ((F5a94@@@@ri   r   r]   r\   c               H    t          | j                  } || |||          S )Nr?  )r=  rd   )r[   r   r]   r^   r@  s        rg   r   r     s,     6<((E5didCCCCri   r   c                   | j         |k    rt          d| j          d| d          t          | j                  sw| j         |k    rt          d| j          d| d          t	          |          t	          |           k    r/t          dt	          |            dt	          |                     dS |dk    r"t	          |          d	k    rt          d
          dS dS )aV  
    ndim inference and validation.

    Validates that values.ndim and ndim are consistent.
    Validates that len(values) and len(placement) are consistent.

    Parameters
    ----------
    values : array-like
    placement : BlockPlacement
    ndim : int

    Raises
    ------
    ValueError : the number of dimensions do not match
    z0Wrong number of dimensions. values.ndim > ndim [z > ]z1Wrong number of dimensions. values.ndim != ndim [z != zWrong number of items passed z, placement implies r   r   zneed to splitN)r]   r  r   rd   r   )r[   r   r]   s      rg   
check_ndimrD    s-   $ {TVv{VVtVVV
 
 	
 v|,, *;$A(.A A9=A A A   y>>S[[((6F 6 6%(^^6 6   )(
 
s9~~**))) 
**ri   r  !tuple[ArrayLike, DtypeObj | None]c                    t          | t                    r0|                                 } |r|dk    rt          j        |           } t          |t
                    r|j        }| |fS )zU
    Ensure that we don't allow NumpyExtensionArray / NumpyEADtype in internals.
    r   )rx   r*   to_numpyry   
atleast_2dr&   numpy_dtype)r[   rd   r]   s      rg   extract_pandas_arrayrJ  
	  si     &011 +"" 	+D1HH]6**F%&& "!5=ri   r   c                F   |g }t          | t                    rF| D ]B}t          |t                    r|                    |           -|                    |           CnAt          | t                    sJ t          |                       |                    |            |S )z.return a new extended blocks, given the result)rx   r  r   r   rZ   r   )r   r  rs      rg   r  r   	  s    ~&$  	! 	!A!T"" !a    a    		! &%((66$v,,66(fMri   r   c                    | j         |k     r:t          | j                  s&t          d|           } |                     dd          } | S )z:
    Reshape if possible to have values.ndim == ndim.
    z+np.ndarray | DatetimeArray | TimedeltaArrayr   r   )r]   r   rd   r   r   )r[   r]   s     rg   r   r   0	  sM    
 {T"6<00 	+ GPPF^^Ar**FMri   c                0   t          | t          t          f          r|                     t                    S t          | t
          t          f          r| j        } t          | t          j	                  r!| 
                                } d| j        _        n	 | S )a+  
    The array that Series.values returns (public attribute).

    This has some historical constraints, and is overridden in block
    subclasses to return the correct array (e.g. period returns
    object ndarray and datetimetz a datetime64[ns] ndarray instead of
    proper extension array).
    F)rx   r?   r=   r   rX   r;   r@   r  ry   r   r'  flags	writeablerh  s    rg   r   r   @	  s     &;677 !}}V$$$	F]N;	<	< ! &"*%% !& 	Mri   )r[   rL   ra   rL   )rd   rO   ra   r:  rk   )r[   rL   r   r   r^   r   ra   rZ   )r   r   r]   r\   r^   r   ra   rZ   )r   r   r]   r\   ra   r   )r[   rL   rd   r  r]   r\   ra   rE  r  )r   )r[   rL   r]   r\   ra   rL   )
__future__r   r  r7  typingr   r   r   r   r   r	   r  numpyry   pandas._libsr
   r   libinternalsr   pandas._libs.internalsr   r   pandas._libs.missingr   pandas.errorsr   r   r   pandas.util._decoratorsr   pandas.util._exceptionsr   pandas.util._validatorsr   pandas.core.dtypes.astyper   r   pandas.core.dtypes.castr   r   r   r   r   pandas.core.dtypes.commonr   r   r   r    r!   r"   pandas.core.dtypes.dtypesr#   r$   r%   r&   r'   pandas.core.dtypes.genericr(   r)   r*   r+   pandas.core.dtypes.inferencer,   pandas.core.dtypes.missingr-   r.   r/   pandas.corer0   pandas.core.algorithmscore
algorithmsru  pandas.core.array_algos.putmaskr1   r2   r3   r4   r5    pandas.core.array_algos.quantiler6   pandas.core.array_algos.replacer7   r8   r9   "pandas.core.array_algos.transformsr:   pandas.core.arraysr;   r<   r=   r>   r?   r@   pandas.core.arrays.string_rA   pandas.core.baserB   pandas.core.commoncommonr  pandas.core.computationrC   pandas.core.constructionrD   rE   pandas.core.indexersrF   pandas.core.indexes.baserG   collections.abcrH   rI   rJ   rK   pandas._typingrL   rM   rN   rO   rP   rQ   rR   rS   rT   rU   pandas.core.apirV   pandas.core.arrays._mixinsrW   rd   rl   rZ   r  rw  r*  r  r5  r   r=  r  r   rD  rJ  r  r   r   r`   ri   rg   <module>rx     s   " " " " " "  				                             
        $ # # # # #         
 3 2 2 2 2 2 4 4 4 4 4 4 7 7 7 7 7 7                                                            / . . . . .                 & & & & & & & & &              = < < < < <         
 5 4 4 4 4 4                3 2 2 2 2 2 ) ) ) ) ) )                   / / / / / /        7 6 6 6 6 6 7 7 7 7 7 7 G                                   &%%%%%FFFFFF RXh
|( |( |( |( |(L,, |( |( |(~-A8 A8 A8 A8 A8E A8 A8 A8HG G G G G] G G GT       45 5 5 5 5- 5 5 5+ + + + +3 + + +   :   < RV
A 
A 
A 
A 
A$ $(D D D D D D&* &* &* &*R   ,               ri   