U
|
¬ý°d0 ã @ sÆ d dl mZ d dlmZ d dlZd dlZd dlZd dlZd dlZd dl m
|
Z
|
m Z m Z m Z mZmZmZ d dlZd dlZd dlmZ d dlmZmZmZmZmZm Z m Z m Z m Z m Z m!Z!m"Z"m#Z# d dl$m%Z%m&Z&m'Z'm(Z( d dl)m*Z* d d l+m,Z, d d
|
l-m.Z.m/Z/m0Z0m1Z1m2Z2m3Z3m4Z4 d d l5m6Z6 d d l7m8Z8 d d l9m:Z: d dl;m<Z<m=Z= d dl>m? m@ZA d dlBmCZCmDZDmEZE d dlFmGZG d dlHmIZI e%shd dlJZKd dlLmMZN d dlOmPZP d dlQmRZR eNjSeNjTeNjUeNjVeNjWeNjXdZYeNjZdd eNj[dd eNj\dd dZ]dddd d d Z^d d dd d!d"Z_eNj`d#d eNjad$d eNjbd%d d&d d'd d(d d)d ececececeNjdd*d d+Zee
|
rd d,lmfZfmgZg d d-lhmiZi ed.d/d0Zjd1d2 Zkd3d4d5d6d7ZlG d8d/ d/e:e=eGZmdS )9é )Ú annotations)ÚdeepcopyN)Ú TYPE_CHECKINGÚAnyÚCallableÚLiteralÚSequenceÚTypeVarÚcast)Úlib) Ú ArrayLikeÚAxisIntÚDtypeÚ FillnaOptionsÚIteratorÚNpDtypeÚPositionalIndexerÚScalarÚSortKindÚ TakeIndexerÚ TimeAmbiguousÚTimeNonexistentÚnpt)Úpa_version_under7p0Úpa_version_under8p0Úpa_version_under9p0Úpa_version_under11p0)Údoc)Úvalidate_fillna_kwargs)Ú is_array_likeÚ is_bool_dtypeÚ
|
is_integerÚis_integer_dtypeÚ is_list_likeÚis_object_dtypeÚ is_scalar)Úisna)Ú roperator)ÚOpsMixin)ÚExtensionArrayÚ ExtensionArraySupportsAnyAll)Úcheck_array_indexerÚunpack_tuple_and_ellipsesÚvalidate_indices)ÚBaseStringArrayMethods)Ú to_offset)Úfallback_performancewarning)Ú
|
ArrowDtype)ÚeqÚneÚltÚgtÚleÚgec C s t || ¡S ©N)ÚpcÚ
|
and_kleene©ÚxÚy© r> úUd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/core/arrays/arrow/array.pyÚ<lambda>\ ó r@ c C s t || ¡S r8 )r9 Ú or_kleener; r> r> r? r@ ^ rA c C s t || ¡S r8 )r9 Úxorr; r> r> r? r@ ` rA )Úand_Zrand_Úor_Zror_rC Zrxorúpa.ChunkedArrayzpa.Array | pa.Scalar)Ú arrow_arrayÚ pa_objectÚreturnc C s. t j | j¡r*t j |j¡r*| t ¡ ¡S | S r8 )ÚpaÚtypesr! Útyper
|
Úfloat64)rG rH r> r> r? Úcast_for_truedivc s
|
ÿrN z&pa.ChunkedArray | pa.Array | pa.Scalar)ÚleftÚrightrI c C s> t t | |¡¡}tj | j¡r:tj |j¡r:| | j¡}|S r8 )r9 ÚfloorÚdividerJ rK r! rL r
|
)rO rP Úresultr> r> r? Úfloordiv_compatn s rT c C s t || ¡S r8 )r9 Ú add_checkedr; r> r> r? r@ { rA c C s t || ¡S r8 )r9 Úsubtract_checkedr; r> r> r? r@ } rA c C s t || ¡S r8 )r9 Úmultiply_checkedr; r> r> r? r@ rA c C s t t| ||¡S r8 ©r9 rR rN r; r> r> r? r@ rA c C s t |t| |¡S r8 rX r; r> r> r? r@ rA c C s
|
t | |S r8 ©rT r; r> r> r? r@ rA c C s
|
t || S r8 rY r; r> r> r? r@ rA c C s t || ¡S r8 )r9 Ú power_checkedr; r> r> r? r@ rA )ÚaddÚraddÚsubZrsubÚmulZrmulÚtruedivZrtruedivÚfloordivZ rfloordivÚmodZrmodÚdivmodZrdivmodÚpowZrpow)Ú NumpySorterÚNumpyValueArrayLike)ÚSeriesÚArrowExtensionArrayTÚArrowExtensionArray)Úboundc C s: t r4t| dd¡d d d
}|dkr0t| |S | jS )Nú[é éÿÿÿÿ)ÚsÚmsÚusÚns)r ÚstrÚsplitÚ
|
ValueErrorÚunit)Úpa_dtypert r> r> r? Úget_unit_from_pa_dtype s rv z'ArrowDtype | pa.DataType | Dtype | Nonezpa.DataType | None©ÚdtyperI c C sL t | tr| jS t | tjr | S | rHz t | ¡W S tjk
|
rF Y nX dS )z3
|
Convert dtype to a pyarrow type instance.
|
N)Ú
|
isinstancer1 Ú pyarrow_dtyperJ ZDataTypeZfrom_numpy_dtypeÚArrowNotImplementedError©rx r> r> r? Úto_pyarrow_type¡ s
|
r} c sF e Zd ZU dZded< ded< dddd d
|
Zed d d ddd ddZed d d ddd ddZddddZ ddddZ
|
dtd d Z dud d d d!d"Z d#d#d$d%d&Z d#d#d$d'd(Zd#d#d$d)d*Zd#d#d$d+d,Zd-d. Zddd/d0Zd1d2 Zd3d4 Zd5d6 Zd7d8 Zddd9d:Zeddd;d<Zed=dd>d?Zd=dd@dAZdd fdBdC Z edddDdEZ dFddGdHZ dIdJddJdKdLZ dIdJddJdMdNZ dIdOdPdQddRdSd dTdUdVZ!ddSd=dW fdXdY Z"dvdd=dZd[d\Z#dwdd=dZd]d^Z$d#d#d$d_d`Z%d#d#d$dadbZ&e'e(j)dxd#dcddded#df fdgdh Z)dFddidjZ*dkddldmZ+e'e(j,dyddndodpdqZ,drds Z-dzd#d=d#dudvdwZ.e'e(j/d{dydzd{d|d}d~dZ/d|dddd dddZ0e'e(j1d d e2j3fd
ddd dddZ1d#d#d$ddZ4d}dddddZ5edd#dddZ6dIdJdSddd fddZ7dIdJdSdd fddZ8ddddZ9dtdddId dd dSdSddd fd¡d¢Z:d#d£dSd#d¤d¥d¦Z;d~d#dd#d§d¨d©Z<dªd« Z=ed¬ddd®d¯d°Z>edd¬dd±d²d³Z?d´dµd¶d·d¸Z@ddSd=d¹dºd»ZAdd=d½dSd¾d¿dÀZBddd=ddÁdÂdÃZCddSdÄdÅdÆZDddSdÄdÇdÈZEddÊdËd=dd=ddÌdÍdÎZFdÏdÐdÑdÒZGd
dSdd=dÓdÔdÕdÖZHddd=dÓd×dØdÙZIddSd=dedÚdÛdÜZJd=dÝdÞdßZKdSdàdádâZLdSddãdädåZMdSddãdædçZNddedededèdédêZOddededëdìdídîZPdïdð ZQdñdò ZRdódô ZSdõdö ZTd÷dø ZUdùdú ZVdûdü ZWdýdþ ZXdÿd ZYdd ZZdd Z[dd Z\dd Z]d d
|
Z^d d Z_dd dZ`dddZadddZbdSdddZcdSdddZddd ZeddSdSd d d ZfddSd=dd d d!ZgddSd=d¹d"d#ZhddSdàd%d&ZiddSd=dedÚd'd(ZjddSd=dedÚd)d*ZkdSd+d,d-ZlddSd=d.d/d0Zmddëdedd1d2d3d4Znddëded5d6d7Zod8d9d:d;Zpd=d<d=d>Zqed?d@ ZredAdB ZsedCdD ZtetZuetZvedEdF ZwewZxedGdH ZydIdJ ZzedKdL Z{edMdN Z|edOdP Z}edQdR Z~edSdT ZedUdV ZedWdX ZedYdZ Zed[d\ Zed]d^ ZdSd_d`daZ
ddcdddedfdgdhZddddedidjdkZddddedidldmZddddedidndoZdpdq ZddddedidrdsZ ZS ( rh aW
|
Pandas ExtensionArray backed by a PyArrow ChunkedArray.
|
|
.. warning::
|
|
ArrowExtensionArray is considered experimental. The implementation and
|
parts of the API may change without warning.
|
|
Parameters
|
----------
|
values : pyarrow.Array or pyarrow.ChunkedArray
|
|
Attributes
|
----------
|
None
|
|
Methods
|
-------
|
None
|
|
Returns
|
-------
|
ArrowExtensionArray
|
|
Notes
|
-----
|
Most methods are implemented using `pyarrow compute functions. <https://arrow.apache.org/docs/python/api/compute.html>`__
|
Some methods may either raise an exception or raise a ``PerformanceWarning`` if an
|
associated compute function is not available based on the installed version of PyArrow.
|
|
Please install the latest version of PyArrow to enable the best functionality and avoid
|
potential bugs in prior versions of PyArrow.
|
|
Examples
|
--------
|
Create an ArrowExtensionArray with :func:`pandas.array`:
|
|
>>> pd.array([1, 1, None], dtype="int64[pyarrow]")
|
<ArrowExtensionArray>
|
[1, 1, <NA>]
|
Length: 3, dtype: int64[pyarrow]
|
rF Ú_datar1 Ú_dtypezpa.Array | pa.ChunkedArrayÚNone)ÚvaluesrI c C sf t rd}t|t|tjr,t |g¡| _n(t|tjr@|| _ntdt | dt
|
| jj | _ d S )NzBpyarrow>=7.0.0 is required for PyArrow backed ArrowExtensionArray.zUnsupported type 'z' for ArrowExtensionArray) r Ú ImportErrorry rJ ÚArrayÚ chunked_arrayr~ Ú ChunkedArrayrs rL r1 r )Úselfr Úmsgr> r> r? Ú__init__æ s ÿz ArrowExtensionArray.__init__NF©rx Úcopyz Dtype | NoneÚboolc C sº t |}t|| r|j}n`t|tjtjfsz|r@t|r@t|}ztj||dd}W n$ tj k
|
rx tj|dd}Y nX |r|
|
|¡}| |}tj |j ¡r¶|jdkr¶| |jj¡}|S )zL
|
Construct a new ExtensionArray from a sequence of scalars.
|
T©rL Ú from_pandas©r r )r} ry r~ rJ r r
r r ÚarrayÚ ArrowInvalidr
|
rK Ú is_durationrL Ú
|
null_countÚfillnarx Úna_value)ÚclsÚscalarsrx r ru Úarrr> r> r? Ú_from_sequenceô s
|
|
z"ArrowExtensionArray._from_sequencec C sÔ t |}|dks(tj |¡s(tj |¡r0|}ntj |¡rXddlm} ||dd}nltj |¡rddlm} ||ddj }nBtj
|
|¡r ddl m } ||dd}|j dkrÄt|}t|tjtjfsâtj|t ¡ dd }t |d|¡}z| t ¡ ¡}W n tjk
|
r Y nX n¦tj |¡rFdd
|
lm} | |d d}n~tj |¡rpdd l m }
|
|
|
|¡ ¡ }nTtj !|¡stj "|¡stj #|¡r´dd l$m%} | |dd}nt&d| d| j'|||dS )zL
|
Construct a new ExtensionArray from a sequence of strings.
|
Nr )Ú to_datetimeÚraise)Úerrors)Ú to_timedeltarp Tr )Úto_timeZcoerce)Ú BooleanArray)Ú
|
to_numericzConverting strings to z is not implemented.r )(r} rJ rK Ú is_binaryÚ is_stringZ is_timestampZpandas.core.tools.datetimesr Úis_dateÚdater Z pandas.core.tools.timedeltasr rt r& ry r r
r Ústringr9 Úif_elser
|
Úint64r Úis_timeZpandas.core.tools.timesr Ú
|
is_booleanZpandas.core.arraysr Ú_from_sequence_of_stringsÚto_numpyr! Ú is_floatingÚ
|
is_decimalZpandas.core.tools.numericr ÚNotImplementedErrorr ) r Ústringsrx r Úpa_typer r r Úmaskr r r r> r> r? r© sX ÿ
|
þ
|
ý
|
ÿ
|
þ
|
ý
|
ÿz-ArrowExtensionArray._from_sequence_of_stringsr )Úitemc C s t | |}t|tjrt|s^| jjdkr@| jjdkr@t ¡ }n| jj
|
}t | tj g |dS t |jrr| |¡S t|jrt | | j |¡S tdnt|tr¬t|}|tkr¼td}t|rÔt|sÔtd| j| }t|tjröt | |S | ¡ }|dkr| jj S |S dS )a8 Select a subset of self.
|
|
Parameters
|
----------
|
item : int, slice, or ndarray
|
* int: The position in 'self' to get.
|
* slice: A slice object, where 'start', 'stop', and 'step' are
|
integers or None
|
* ndarray: A 1-d boolean NumPy ndarray the same length as 'self'
|
|
Returns
|
-------
|
item : scalar or ExtensionArray
|
|
Notes
|
-----
|
For scalar ``item``, return a scalar value suitable for the array's
|
type. This should be an instance of ``self.dtype.type``.
|
For slice ``key``, return an instance of ``ExtensionArray``, even
|
if the slice is length 0 or 1.
|
For a boolean mask, return an instance of ``ExtensionArray``, filtered
|
to the values where ``item`` is True.
|
r¤ Úpyarrow©rL zFOnly integers, slices and integer or boolean arrays are valid indices.Nzuonly integers, slices (`:`), ellipsis (`...`), numpy.newaxis (`None`) and integer or boolean arrays are valid indices) r+ ry ÚnpÚndarrayÚlenr ÚnameZstoragerJ r¤ rz rL r r" rx Útaker r~ ÚfilterÚ
|
IndexErrorÚtupler, ÚEllipsisÚslicer% r! r
Úas_pyr )r r± ru ÚvalueÚscalarr> r> r? Ú __getitem__N s:
|
|
|
|
|
ÿ
|
ÿ
|
|
z ArrowExtensionArray.__getitem__z Iterator[Any]©rI c c s6 | j j}| jD ]"}| ¡ }|dkr*|V q|V qdS )z5
|
Iterate over elements of the array.
|
N)r r r~ r¾ )r r r¿ Úvalr> r> r? Ú__iter__ s
|
z ArrowExtensionArray.__iter__c C s | j S )z)Convert myself to a pyarrow ChunkedArray.)r~ )r rL r> r> r? Ú__arrow_array__ s z#ArrowExtensionArray.__arrow_array__zNpDtype | Nonez
|
np.ndarrayrw c C s | j |dS )z?Correctly construct numpy arrays when passed to `np.asarray()`.r| ©rª )r rx r> r> r? Ú __array__£ s z ArrowExtensionArray.__array__rg )r rI c C s t | t | j¡S r8 )rL r9 Úinvertr~ ©r r> r> r? Ú
|
__invert__§ s z ArrowExtensionArray.__invert__c C s t | t | j¡S r8 )rL r9 Znegate_checkedr~ rÉ r> r> r? Ú__neg__ª s zArrowExtensionArray.__neg__c C s t | | jS r8 ©rL r~ rÉ r> r> r? Ú__pos__ s zArrowExtensionArray.__pos__c C s t | t | j¡S r8 )rL r9 Z abs_checkedr~ rÉ r> r> r? Ú__abs__° s zArrowExtensionArray.__abs__c C s | j ¡ }| j ¡ |d< |S ©Nr~ )Ú__dict__r r~ Úcombine_chunks©r Ústater> r> r? Ú __getstate__µ s
|
z ArrowExtensionArray.__getstate__c C s" t |d ¡|d< | j |¡ d S rÏ )rJ r rÐ ÚupdaterÒ r> r> r? Ú __setstate__º s z ArrowExtensionArray.__setstate__c C s0 ddl m} t|j }t|tr0|| j|j}nøt|tjt frN|| j|}nÚt||rj|| j|
|
¡ }n¾t |rz|| jt |¡}W n t jjt jjfk
|
r t| t|B }| }tjt| dd}|t | ¡| |||< t j|t ¡ d}t ||d ¡}Y nX nt|j dt| t|S )Nr ©ÚBaseMaskedArrayr r| r³ ú not implemented for )Úpandas.core.arrays.maskedrØ ÚARROW_CMP_FUNCSÚ__name__ry rh r~ r´ rµ ÚlistrÅ r% rJ rÀ r r{ r r& Úzerosr¶ r Úbool_r9 r¥ r rL )r ÚotherÚoprØ Úpc_funcrS r° Zvalidr> r> r? Ú _cmp_method¾ s,
|
|
|
|
ÿz ArrowExtensionArray._cmp_methodc C s¾ ddl m} | jj}tj |¡s,tj |¡rÈ|tj t
|
j fkrÈ| j ¡ }tj |¡r^dg| }n
|
dg| }t |r||g| }nt|t| r|j}|tj kr¬t | j||¡}nt || j|¡}t| |S ||j } | tkrêt|j dt|tr| | j|j}n¬t|tjtfr0| | jtj|dd}nt||rN| | j| ¡ }ndt |rt|r|jtkrtj d | jjd}
|
n
|
t |¡}
|
| | j|
|
}nt|j d t| t| |S )
|
Nr r× Ú rA z not implemented.Tr r³ rÙ ) rÚ rØ r~ rL rJ rK r¡ r Úoperatorr[ r' r\ Úlengthr% ry r9 Zbinary_join_element_wiserÜ ÚNotImplementedr rh r´ rµ rÝ r rÅ r& ÚARROW_LOGICAL_FUNCSrÀ ) r rà rá Z arrow_funcsrØ r¯ ræ ÚsepsrS râ Z pa_scalarr> r> r? Ú_evaluate_op_methodÙ sH þ
|
|
|
|
|
|
ÿz'ArrowExtensionArray._evaluate_op_methodc C s | ||t¡S r8 )rê rè ©r rà rá r> r> r? Ú_logical_method
|
s z#ArrowExtensionArray._logical_methodc C s | ||t¡S r8 )rê ÚARROW_ARITHMETIC_FUNCSrë r> r> r? Ú _arith_method s z!ArrowExtensionArray._arith_methodc C s t |tsdS | j|jkS ©NF)ry rh r~ )r rà r> r> r? Úequals s
|
zArrowExtensionArray.equalsc C s | j S )z2
|
An instance of 'ExtensionDtype'.
|
)r rÉ r> r> r? rx s zArrowExtensionArray.dtypeÚintc C s | j jS )zL
|
The number of bytes needed to store this object in memory.
|
)r~ ÚnbytesrÉ r> r> r? rò s zArrowExtensionArray.nbytesc C s
|
t | jS )z]
|
Length of this array.
|
|
Returns
|
-------
|
length : int
|
)r¶ r~ rÉ r> r> r? Ú__len__% s zArrowExtensionArray.__len__c s\ t |rL|| jjk rL| jjdkrHt |¡rHt |rHt t | j ¡¡
|
¡ S dS t t |¡S )NÚfF)r& rx r Úkindr Zis_floatr9 ÚanyÚis_nanr~ r¾ r ÚsuperÚ __contains__)r Úkey©Ú __class__r> r? rù / s
|
z ArrowExtensionArray.__contains__c C s | j jdkS )Nr )r~ r rÉ r> r> r? Ú_hasna; s zArrowExtensionArray._hasnaznpt.NDArray[np.bool_]c C s | j ¡ ¡ S )z
|
Boolean NumPy array indicating if each value is missing.
|
|
This should return a 1-D array the same length as 'self'.
|
)r~ Úis_nullrª rÉ r> r> r? r& ? s zArrowExtensionArray.isnaT©Úskipnac K s | j dd|i|S )a
|
Return whether any element is truthy.
|
|
Returns False unless there is at least one element that is truthy.
|
By default, NAs are skipped. If ``skipna=False`` is specified and
|
missing values are present, similar :ref:`Kleene logic <boolean.kleene>`
|
is used as for logical operations.
|
|
Parameters
|
----------
|
skipna : bool, default True
|
Exclude NA values. If the entire array is NA and `skipna` is
|
True, then the result will be False, as for an empty array.
|
If `skipna` is False, the result will still be True if there is
|
at least one element that is truthy, otherwise NA will be returned
|
if there are NA's present.
|
|
Returns
|
-------
|
bool or :attr:`pandas.NA`
|
|
See Also
|
--------
|
ArrowExtensionArray.all : Return whether all elements are truthy.
|
|
Examples
|
--------
|
The result indicates whether any element is truthy (and by default
|
skips NAs):
|
|
>>> pd.array([True, False, True], dtype="boolean[pyarrow]").any()
|
True
|
>>> pd.array([True, False, pd.NA], dtype="boolean[pyarrow]").any()
|
True
|
>>> pd.array([False, False, pd.NA], dtype="boolean[pyarrow]").any()
|
False
|
>>> pd.array([], dtype="boolean[pyarrow]").any()
|
False
|
>>> pd.array([pd.NA], dtype="boolean[pyarrow]").any()
|
False
|
>>> pd.array([pd.NA], dtype="float64[pyarrow]").any()
|
False
|
|
With ``skipna=False``, the result can be NA if this is logically
|
required (whether ``pd.NA`` is True or False influences the result):
|
|
>>> pd.array([True, False, pd.NA], dtype="boolean[pyarrow]").any(skipna=False)
|
True
|
>>> pd.array([1, 0, pd.NA], dtype="boolean[pyarrow]").any(skipna=False)
|
True
|
>>> pd.array([False, False, pd.NA], dtype="boolean[pyarrow]").any(skipna=False)
|
<NA>
|
>>> pd.array([0, 0, pd.NA], dtype="boolean[pyarrow]").any(skipna=False)
|
<NA>
|
rö r )rö ©Ú_reduce©r r Úkwargsr> r> r? rö G s 8zArrowExtensionArray.anyc K s | j dd|i|S )a
|
Return whether all elements are truthy.
|
|
Returns True unless there is at least one element that is falsey.
|
By default, NAs are skipped. If ``skipna=False`` is specified and
|
missing values are present, similar :ref:`Kleene logic <boolean.kleene>`
|
is used as for logical operations.
|
|
Parameters
|
----------
|
skipna : bool, default True
|
Exclude NA values. If the entire array is NA and `skipna` is
|
True, then the result will be True, as for an empty array.
|
If `skipna` is False, the result will still be False if there is
|
at least one element that is falsey, otherwise NA will be returned
|
if there are NA's present.
|
|
Returns
|
-------
|
bool or :attr:`pandas.NA`
|
|
See Also
|
--------
|
ArrowExtensionArray.any : Return whether any element is truthy.
|
|
Examples
|
--------
|
The result indicates whether all elements are truthy (and by default
|
skips NAs):
|
|
>>> pd.array([True, True, pd.NA], dtype="boolean[pyarrow]").all()
|
True
|
>>> pd.array([1, 1, pd.NA], dtype="boolean[pyarrow]").all()
|
True
|
>>> pd.array([True, False, pd.NA], dtype="boolean[pyarrow]").all()
|
False
|
>>> pd.array([], dtype="boolean[pyarrow]").all()
|
True
|
>>> pd.array([pd.NA], dtype="boolean[pyarrow]").all()
|
True
|
>>> pd.array([pd.NA], dtype="float64[pyarrow]").all()
|
True
|
|
With ``skipna=False``, the result can be NA if this is logically
|
required (whether ``pd.NA`` is True or False influences the result):
|
|
>>> pd.array([True, True, pd.NA], dtype="boolean[pyarrow]").all(skipna=False)
|
<NA>
|
>>> pd.array([1, 1, pd.NA], dtype="boolean[pyarrow]").all(skipna=False)
|
<NA>
|
>>> pd.array([True, False, pd.NA], dtype="boolean[pyarrow]").all(skipna=False)
|
False
|
>>> pd.array([1, 0, pd.NA], dtype="boolean[pyarrow]").all(skipna=False)
|
False
|
Úallr )r r r r> r> r? r s 8zArrowExtensionArray.allZ quicksortÚlast)Ú ascendingrõ Ú na_positionr rq )r rõ r rI c K s^ |rdnd}ddd |d ¡}|d kr4td| tj| j||d}| ¡ }|jtjdd S )
|
Nr Ú
|
descendingÚat_endÚat_start)r Úfirstzinvalid na_position: )ÚorderÚnull_placementF©r ) Úgetrs r9 Úarray_sort_indicesr~ rª Úastyper´ Úintp) r r rõ r r r r rS Z np_resultr> r> r? Úargsort» s ÿzArrowExtensionArray.argsort)r ÚmethodrI c s| | j ¡ d| j jfks | jr8|s8tt d| |dS | j }tj |j ¡rZ|
|
t ¡ ¡}tt |||d}t ||¡ ¡ S )Nr Úargrÿ )Ú
|
skip_nulls)r~ ræ r rý Úgetattrrø rJ rK r rL r
|
r¦ r9 Úindexr¾ )r r r Údatar¿ rû r> r? Ú _argmin_maxÎ s ÿÿz ArrowExtensionArray._argmin_max)r rI c C s | |d¡S )NÚmin©r ©r r r> r> r? ÚargminÞ s zArrowExtensionArray.argminc C s | |d¡S )NÚmaxr r r> r> r? Úargmaxá s zArrowExtensionArray.argmaxc C s t | | jS )z·
|
Return a shallow copy of the array.
|
|
Underlying ChunkedArray is immutable, so a deep copy is unnecessary.
|
|
Returns
|
-------
|
type(self)
|
rÌ rÉ r> r> r? r ä s
|
zArrowExtensionArray.copyc C s t | t | j¡S )z|
|
Return ArrowExtensionArray without NA values.
|
|
Returns
|
-------
|
ArrowExtensionArray
|
)rL r9 Ú drop_nullr~ rÉ r> r> r? Údropnað s zArrowExtensionArray.dropnazobject | ArrayLike | NonezFillnaOptions | Nonez
|
int | None)r r¿ r ÚlimitrI c s0 t ||\}}|d k r(t j|||dS |d k rHt t j|||dS t|rtt|}t|t| krtdt| dt| dd }||| j j
|
| j }zd|d krÄt
|
| t j | j |dW S |dkrât
|
| t | j ¡W S |dkrt
|
| t | j ¡W S W n tjk
|
r Y nX t j|||dS ) N)r¿ r r$ z'Length of 'value' does not match. Got (z ) expected c
|
S s | d kr | S t | tjtjtjfr&| S t| r6tj}ntj}z|| |dd} W nB tjk
|
r } z"dt | d| }t
|
||W 5 d }~X Y nX | S )NTr úInvalid value 'ú ' for dtype ) ry rJ r r r
r r rÀ ÚArrowTypeErrorrq Ú TypeError)r¿ r¯ rx Úpa_boxÚerrr r> r> r? Úconvert_fill_value s z6ArrowExtensionArray.fillna.<locals>.convert_fill_value)Ú
|
fill_valueÚpadZbackfill)r rø r r0 r r
|
r r¶ rs r~ rL rx r9 Ú fill_nullZfill_null_forwardZfill_null_backwardrJ r{ )r r¿ r r$ r+ r, rû r> r? r ú s0
|
ÿ
|
zArrowExtensionArray.fillnac C sD t |stjt | tdS tj| jtj|ddd}tj|tj dS )Nr| Tr )Z value_set)
|
r¶ r´ rÞ r r9 Úis_inr~ rJ r rß )r r rS r> r> r? Úisin4 s zArrowExtensionArray.isinztuple[np.ndarray, Any]c C s | j ¡ }|| jjfS )a+
|
Return an array and missing value suitable for factorization.
|
|
Returns
|
-------
|
values : ndarray
|
na_value : pd.NA
|
|
Notes
|
-----
|
The values returned by this method are also used in
|
:func:`pandas.util.hash_pandas_object`.
|
)r~ rª rx r )r r r> r> r? Ú_values_for_factorize> s
|
z)ArrowExtensionArray._values_for_factorizez!tuple[np.ndarray, ExtensionArray])Úuse_na_sentinelrI c C sð |rdnd}| j j}tj |¡r2| j t ¡ ¡}n| j }|j|d}| ¡ dkr|t j
|
g t j d}t| tj g |jj d}nN| ¡ j}|jdkrt |d¡}|jdd d
|
jt j dd }t| | d¡j}tj |¡rètt| | j¡}||fS ) Nr° Úencode)Ú null_encodingr r| r³ rl FT)Úzero_copy_onlyÚwritabler )r~ rL rJ rK r r
|
r¦ Zdictionary_encoderæ r´ r r r Z
|
value_typerÑ Úindicesr r9 r. rª r ÚchunkÚ
|
dictionaryrh rx ) r r2 r4 r¯ r Úencodedr7 ÚuniquesZ
|
pa_indicesr> r> r? Ú factorizeO s(
|
|
ÿ z ArrowExtensionArray.factorizec O s t t| dd S )NzA does not support reshape as backed by a 1D pyarrow.ChunkedArray.)r rL )r Úargsr r> r> r? Úreshapen s ÿzArrowExtensionArray.reshaper )r ÚdecimalsrI c O s t | tj| j|dS )a
|
Round each value in the array a to the given number of decimals.
|
|
Parameters
|
----------
|
decimals : int, default 0
|
Number of decimal places to round to. If decimals is negative,
|
it specifies the number of positions to the left of the decimal point.
|
*args, **kwargs
|
Additional arguments and keywords have no effect.
|
|
Returns
|
-------
|
ArrowExtensionArray
|
Rounded values of the ArrowExtensionArray.
|
|
See Also
|
--------
|
DataFrame.round : Round values of a DataFrame.
|
Series.round : Round values of a Series.
|
)Úndigits)rL r9 Úroundr~ )r r? r= r r> r> r? rA t s zArrowExtensionArray.roundrO z$NumpyValueArrayLike | ExtensionArrayzLiteral[('left', 'right')]rd z npt.NDArray[np.intp] | np.intp)r¿ ÚsideÚsorterrI c C s6 | j rtdt|tr"| t¡}| ¡ j|||dS )NzOsearchsorted requires array to be sorted, which is impossible with NAs present.)rB rC )rý rs ry r) r Úobjectrª Ú searchsorted)r r¿ rB rC r> r> r? rE s ÿ
|
|
z ArrowExtensionArray.searchsortedr r )r7 Ú
|
allow_fillr, rI c C s( t |st |¡}n|}t| jdkr:|dk ¡ r:td|jdkr^| ¡ t| jkr^td|rÞ|dk }| ¡ rÈt |t| j t
|
j ||d}| j |¡}t |r°t| |S t| |}|||< |S t| | j |¡S nF|dk ¡ rt |¡}||dk t| j7 < t| | j |¡S dS )a
|
Take elements from an array.
|
|
Parameters
|
----------
|
indices : sequence of int or one-dimensional np.ndarray of int
|
Indices to be taken.
|
allow_fill : bool, default False
|
How to handle negative values in `indices`.
|
|
* False: negative values in `indices` indicate positional indices
|
from the right (the default). This is similar to
|
:func:`numpy.take`.
|
|
* True: negative values in `indices` indicate
|
missing values. These values are set to `fill_value`. Any other
|
other negative values raise a ``ValueError``.
|
|
fill_value : any, optional
|
Fill value to use for NA-indices when `allow_fill` is True.
|
This may be ``None``, in which case the default NA value for
|
the type, ``self.dtype.na_value``, is used.
|
|
For many ExtensionArrays, there will be two representations of
|
`fill_value`: a user-facing "boxed" scalar, and a low-level
|
physical NA value. `fill_value` should be the user-facing version,
|
and the implementation should handle translating that to the
|
physical version for processing the take if necessary.
|
|
Returns
|
-------
|
ExtensionArray
|
|
Raises
|
------
|
IndexError
|
When the indices are out of bounds for the array.
|
ValueError
|
When `indices` contains negative values other than ``-1``
|
and `allow_fill` is True.
|
|
See Also
|
--------
|
numpy.take
|
api.extensions.take
|
|
Notes
|
-----
|
ExtensionArray.take is called by ``Series.__getitem__``, ``.loc``,
|
``iloc``, when `indices` is a sequence of values. Additionally,
|
it's called by :meth:`Series.reindex`, or any other method
|
that causes realignment, with a `fill_value`.
|
r zcannot do a non-empty takez!out of bounds value in 'indices'.)r° N)r r´ Z
|
asanyarrayr¶ r~ rö rº Úsizer r- rJ r r¸ r& rL r )r r7 rF r, Z indices_arrayZ fill_maskrS r> r> r? r¸ s. <
|
zArrowExtensionArray.takeznpt.DTypeLike | NonerD )rx r r rI c C s0 |d kr| j rt}|tjkr$| jj}| jj}tj
|
|¡rXtj |¡sXt j t| |d}nÀt|r| j rt jt| td}| ¡ }t | | j¡||< ntj | jj¡rÐt j| j|d}t|sÌ||d d
< |S | j rú| ¡ }||| ¡ < t j|j|dS t j| j|d}|r| ¡ }| j r,||| ¡ < |S ©Nr| )rý rD r Ú
|
no_defaultrx r r~ rL rJ rK Ú is_temporalr¢ r´ r rÝ r$ Úemptyr¶ r& Úasarrayrþ r )r rx r r r¯ rS r° r r> r> r? rª s4
|
|
z ArrowExtensionArray.to_numpyc C sX | j j}tj |¡r&| j t ¡ ¡}n| j }t |¡}tj |¡rL| |¡}t| |S )z
|
Compute the ArrowExtensionArray of unique values.
|
|
Returns
|
-------
|
ArrowExtensionArray
|
) r~ rL rJ rK r r
|
r¦ r9 Úunique)r r¯ r Z pa_resultr> r> r? rM ( s
|
|
zArrowExtensionArray.uniquerf )r# rI c C sÀ | j j}tj |¡r&| j t ¡ ¡}n| j }ddlm}m } |
|
¡ }| d¡}| d¡}|r|j dkr| ¡ } | | ¡}| | ¡}tj |¡r| |¡}t|}|t| |}
|
|||
|
dddS )a@
|
Return a Series containing counts of each unique value.
|
|
Parameters
|
----------
|
dropna : bool, default True
|
Don't include counts of missing values.
|
|
Returns
|
-------
|
counts : Series
|
|
See Also
|
--------
|
Series.value_counts
|
r )ÚIndexrf rk ÚcountF)r r· r )r~ rL rJ rK r r
|
r¦ ÚpandasrN rf Ú value_countsÚfieldr Zis_validr¹ rh ) r r# r¯ r rN rf Zvcr Úcountsr° r r> r> r? rQ ? s"
|
|
|
|
|
z ArrowExtensionArray.value_countsztype[ArrowExtensionArrayT])r rI c C sH dd |D }|d j dkr&t ¡ }n |d j j}tj||d}| |S )zÕ
|
Concatenate multiple ArrowExtensionArrays.
|
|
Parameters
|
----------
|
to_concat : sequence of ArrowExtensionArrays
|
|
Returns
|
-------
|
ArrowExtensionArray
|
c S s g | ]}|j ¡ D ]}|qqS r> ©r~ Z
|
iterchunks)Ú.0Zear r> r> r? Ú
|
<listcomp>} s z9ArrowExtensionArray._concat_same_type.<locals>.<listcomp>r r¤ r³ )rx rJ r¤ rz r )r Z to_concatÚchunksru r r> r> r? Ú_concat_same_typen s
|
z%ArrowExtensionArray._concat_same_typez$ArrowExtensionArray | ExtensionArray)r· r rI c s ddi ||¡}tt|d}|dkr<t j|fd|i|S | j}|j}tj |¡rb|
|
t ¡ ¡}||fd|i|}tj |¡r|
|
|¡}t| |S )a¶
|
Return an ExtensionArray performing an accumulation operation.
|
|
The underlying data type might change.
|
|
Parameters
|
----------
|
name : str
|
Name of the function, supported values are:
|
- cummin
|
- cummax
|
- cumsum
|
- cumprod
|
skipna : bool, default True
|
If True, skip NA values.
|
**kwargs
|
Additional keyword arguments passed to the accumulation function.
|
Currently, there is no supported kwarg.
|
|
Returns
|
-------
|
array
|
|
Raises
|
------
|
NotImplementedError : subclass does not define accumulations
|
ZcumsumZcumulative_sum_checkedNr r ) r r r9 rø Ú _accumulater~ rL rJ rK r r
|
r¦ ) r r· r r Ú pyarrow_nameÚ pyarrow_methZ data_to_accumru rS rû r> r? rY s" ÿ þ
|
z ArrowExtensionArray._accumulate)r· r c sü j j} j }|dkr|tj |¡sFtj |¡sFtj |¡sFtj |¡r|tj |¡rd j t ¡ ¡}n j }t
|
|d¡}|}nj|dkr¢tj |¡r¢ j t ¡ ¡}nD|dkrætj |¡ræ|j }|dkrÖ j t ¡ ¡}n j t ¡ ¡}|dkrü fdd} nDd d
|
d d d ||¡}
|
tt
|
|
|
d} | dkr@t j|fd|i|S |dkr^d|kr^d|d< n|dkrpd|d< z| |fd|i|} W nb tttfk
|
rê } z<dt j d j d| dtj d| d } t| | W 5 d} ~ X Y nX |dkrþ| d } t
|
| ¡ ¡ r jjS |dkr8tj |¡r8| |¡} |dkrZtj |¡rZ| |¡} |d krôtj |¡rô| t ¡ ¡} tj |¡r| |¡} nZtj |¡rÂt |}| t |¡¡} n2tj |¡râ| t d ¡¡} n| t |j ¡¡} | ¡ S ) a
|
Return a scalar result of performing the reduction operation.
|
|
Parameters
|
----------
|
name : str
|
Name of the function, supported values are:
|
{ any, all, min, max, sum, mean, median, prod,
|
std, var, sem, kurt, skew }.
|
skipna : bool, default True
|
If True, skip NaN values.
|
**kwargs
|
Additional keyword arguments passed to the reduction function.
|
Currently, `ddof` is the only supported kwarg.
|
|
Returns
|
-------
|
scalar
|
|
Raises
|
------
|
TypeError : subclass does not define reductions
|
)rö r r )r r Úsum)ÚmedianÚmeanÚstdÚsemé r` c s4 t j| fd|i|}t t j¡¡}t ||¡S )Nr )r9 ÚstddevZ sqrt_checkedrO r~ Zdivide_checked)r r r Ú numeratorÚ denominatorrÉ r> r? r[ ñ s z1ArrowExtensionArray._reduce.<locals>.pyarrow_methÚquantileÚproductrb Zvariance)r] Úprodr_ ÚvarNr Z min_countr] g à?Úqr ú'z ' with dtype z does not support reduction 'z' with pyarrow version z. 'z(' may be supported by upgrading pyarrow.)r] r^ )r_ r` rm )!r~ rL rJ rK r! r« r r¬ r
|
r¦ r9 Ú not_equalrJ Ú bit_widthÚint32r r rø r ÚAttributeErrorr r( rÜ rx Ú __version__rþ r¾ r r§ rv Údurationr¢ rt )r r· r r r¯ Zdata_to_reduceZ data_to_cmpZnot_eqÚnbitsr[ rZ rS r* r rt rû rÉ r? r ¸ s
|
ÿ
|
þ
|
ý
|
ü ü û
|
|
|
,ÿ
|
|
|
zArrowExtensionArray._reducec C s2 t |tr t|dkr |d }t| |}| |¡}t |¡rP| d|| j¡}n¾t |rt
|
t |}t| }|dk r|||7 }d| kr|k s¦n t d| d| t |r¸tdnt |tjrÌ| ¡ }| jd|
jtj|g| jjddf| j|d d
j}t |¡ ¡ }nøt|rBtj|tjd }| | j||¡}nÌt |sZt |tjrtj t| tjd }d||< | ||| j¡}nt t| ¡| }t|t|kr´tdt|dkrÆdS t |¡}|| }| |¡}tj t| tjd }d||< | | j||¡}t |tj!r(t |g¡}|| _dS )
|
a÷ Set one or more values inplace.
|
|
Parameters
|
----------
|
key : int, ndarray, or slice
|
When called from, e.g. ``Series.__setitem__``, ``key`` will be
|
one of
|
|
* scalar int
|
* ndarray of integers.
|
* boolean ndarray
|
* slice object
|
|
value : ExtensionDtype.type, Sequence[ExtensionDtype.type], or object
|
value or values to be set of ``key``.
|
|
Returns
|
-------
|
None
|
rk r Tzindex z' is out of bounds for axis 0 with size z%Length of indexer and values mismatchNr r| )"ry r» r¶ r+ Ú _maybe_convert_setitem_valueÚcomZ is_null_sliceÚ_if_elser~ r! r
|
rñ rº r# rs rJ r r¾ rW r rL r rÑ r r´ rL rß Ú_replace_with_maskr% rÞ Úaranger r¸ r ) r rú r¿ r ÚnrW r° r7 r r> r> r? Ú __setitem__0 s\
|
|
|
|
|
ÿ
|
þý
|
|
|
z ArrowExtensionArray.__setitem__ÚaverageÚkeep©Úaxisr Ú na_optionr Úpctr c s~ t s |dkr\t j|||||d}|dks.|r8t ¡ }nt ¡ }tj||dd}t| |S | j ¡ } |rndnd}
|
|dkr~d nd
|
} |dkrd n|} t
|
j | |
|
| | d }|d krÖt
|
| j¡} tj d|jd}t
|
| ||¡}|dkr"t
|
j | |
|
| dd }| t ¡ ¡}| t ¡ ¡}t
|
t
|
||¡d¡}|rrtj |j¡sF| t ¡ ¡}|dkr\t
|
|¡}n
|
t
|
|¡}t
|
||¡}t| |S )z*
|
See Series.rank.__doc__.
|
r r{ ry Tr r r Útopr r
|
r )Ú sort_keysr Ú
|
tiebreakerrz Nr³ r é Zdense)r rø Ú_rankrJ rM Zuint64r rL r~ rÑ r9 Zrankrþ rÀ r¥ r
|
rR r[ rK r« r rO )r r| r r} r r~ Zrankedr¯ rS r r r r r° ÚnullZ
|
result_maxZ
|
result_minZdivisorrû r> r? r sZ û
|
|
ü
|
ü
|
|
zArrowExtensionArray._rankznpt.NDArray[np.float64])r ÚqsÚ interpolationrI c C s¤ | j j}| j }tj |¡rF|j}|dkr8| t ¡ ¡}n| t ¡ ¡}t j
|
|||d}tj |¡r|j}|dkr| t ¡ ¡}n| t ¡ ¡}| |¡}t| |S )zè
|
Compute the quantiles of self for each quantile in `qs`.
|
|
Parameters
|
----------
|
qs : np.ndarray[float64]
|
interpolation: str
|
|
Returns
|
-------
|
same type as self
|
ra )ri r ) r~ rL rJ rK rJ rl r
|
rm r¦ r9 re )r r
r ru r rq rS r> r> r? Ú _quantile s
|
z ArrowExtensionArray._quantile)r r# rI c
|
C sÀ | j j}tj |¡rX|j}|dkr4| j t ¡ ¡}q^|dkrN| j t ¡ ¡}q^t |n| j }|rj|
|
¡ }t |¡}| d¡ t | d¡t | d¡¡¡¡}tj |¡r´| |¡}t| |S )aT
|
Returns the mode(s) of the ExtensionArray.
|
|
Always returns `ExtensionArray` even if only one value.
|
|
Parameters
|
----------
|
dropna : bool, default True
|
Don't consider counts of NA values.
|
|
Returns
|
-------
|
same type as self
|
Sorted, if possible.
|
ra é@ r rS )r~ rL rJ rK rJ rl r
|
rm r¦ r r" r9 rQ rR r¹ Úequalr )r r# r¯ rq r ÚresÚ most_commonr> r> r? Ú_modeé s$
|
|
|
ÿ
|
zArrowExtensionArray._modec
|
C s |dkr |S t |tjtjtjfr&|S t|r6tj}ntj}z||| jj dd}W nD tj
|
k
|
r } z$dt | d| j }t ||W 5 d}~X Y nX |S )z-Maybe convert value to be pyarrow compatible.NTr r% r& )ry rJ r r r
r# r rÀ r~ rL r' rq rx r( )r r¿ r) r* r r> r> r? rr s z0ArrowExtensionArray._maybe_convert_setitem_valuez npt.NDArray[np.bool_] | boolzArrayLike | Scalar)ÚcondrO rP c C st zt |||¡W S tjk
|
r& Y nX dddd}||\}}||\}}|pT|}t |||¡}tj||ddS )an
|
Choose values based on a condition.
|
|
Analogous to pyarrow.compute.if_else, with logic
|
to fallback to numpy for unsupported types.
|
|
Parameters
|
----------
|
cond : npt.NDArray[np.bool_] or bool
|
left : ArrayLike | Scalar
|
right : ArrayLike | Scalar
|
|
Returns
|
-------
|
pa.Array
|
z%tuple[np.ndarray, pa.DataType | None]r c S sL t | tjtjfr| j}n t | tjr6| j}| ¡ } nd }tj| t d|fS rH )
|
ry rJ r r
rL r r¾ r´ r rD )r¿ r¯ r> r> r? Ú_to_numpy_and_type? s
|
z8ArrowExtensionArray._if_else.<locals>._to_numpy_and_typeTr )r9 r¥ rJ r{ r´ Úwherer ) r r rO rP r Z left_typeZ
|
right_typer¯ rS r> r> r? rt # s
|
z ArrowExtensionArray._if_else)r r° Ú replacementsc C s t |tjr| ¡ }trpt |tjrbt t|d¡}t t|¡||< tj
|
|t ¡ d}| |¡}| |||¡S t |tjrtj |j¡r| ¡ }zt |||¡W S tjk
|
r¸ Y nX t |tjrÖtj
|
|td}nt |tjrê| ¡ }tj
|
|td}|||< tj
|
||jddS )a¿
|
Replace items selected with a mask.
|
|
Analogous to pyarrow.compute.replace_with_mask, with logic
|
to fallback to numpy for unsupported types.
|
|
Parameters
|
----------
|
values : pa.Array or pa.ChunkedArray
|
mask : npt.NDArray[np.bool_] or bool
|
replacements : ArrayLike or Scalar
|
Replacement value(s)
|
|
Returns
|
-------
|
pa.Array or pa.ChunkedArray
|
Nr³ r| Tr )ry rJ r
rÑ r r r´ Úfullr¶ rv r r¦ r¸ rt rK r¨ rL r9 Zreplace_with_maskr{ rD r r¾ )r r r° r r7 rS r> r> r? ru O s,
|
z&ArrowExtensionArray._replace_with_maskr zlist[list[Any]])ÚfuncrI c s fdd| j ¡ D S )zJApply a callable to each element while maintaining the chunking structure.c s& g | ] } fd d|j ddD qS )c s g | ]}|d krd n |qS r8 r> )rU rà ©r r> r? rV s ÿzEArrowExtensionArray._apply_elementwise.<locals>.<listcomp>.<listcomp>F)r5 rÆ )rU r8 r r> r? rV s ü
|
|
þz:ArrowExtensionArray._apply_elementwise.<locals>.<listcomp>rT )r r r> r r? Ú_apply_elementwise s
|
ûz&ArrowExtensionArray._apply_elementwise)ÚpatÚflagsc C s( |rt d|t| t | j|¡S )Nz!count not implemented with flags=)r rL r9 Zcount_substring_regexr~ )r r r r> r> r? Ú
|
_str_count s z ArrowExtensionArray._str_countú z"Literal[('left', 'right', 'both')])ÚwidthrB Úfillcharc C sX |dkrt j}n0|dkr t j}n |dkr0t j}ntd| dt| || j||dS )NrO rP ZbothzInvalid side: z-. Side must be one of 'left', 'right', 'both')r Úpadding)r9 Z utf8_lpadZ utf8_rpadZ utf8_centerrs rL r~ )r r rB r Zpa_padr> r> r? Ú_str_pad s
|
ÿz ArrowExtensionArray._str_pad)Úcaser Úregexc C sT |rt d||r tj}ntj}|| j|| d}t|sH| |¡}t| |S )Nz$contains not implemented with flags=)Z ignore_case)r r9 Zmatch_substring_regexZmatch_substringr~ r& r. rL )r r r r Únar Z pa_containsrS r> r> r? Ú _str_contains¥ s
|
z!ArrowExtensionArray._str_contains)r c C s. t j| j|d}t|s"| |¡}t| |S ©N©Úpattern)r9 Z starts_withr~ r& r. rL ©r r r rS r> r> r? Ú_str_startswith´ s
|
z#ArrowExtensionArray._str_startswithc C s. t j| j|d}t|s"| |¡}t| |S r¡ )r9 Ú ends_withr~ r& r. rL r¤ r> r> r? Ú _str_endswithº s
|
z!ArrowExtensionArray._str_endswithrl zstr | re.Patternzstr | Callable)r Úreplrw r r r c C sR t |tjs t|s |r |r$td|r.tjntj}|| j|||d}t | |S )NzRreplace is not supported with a re.Pattern, callable repl, case=False, or flags!=0)r£ Ú replacementZmax_replacements)
|
ry ÚreÚPatternÚcallabler r9 Zreplace_substring_regexZreplace_substringr~ rL ) r r r¨ rw r r r r rS r> r> r? Ú _str_replaceÀ s ÿz ArrowExtensionArray._str_replacezint | Sequence[int])Úrepeatsc C sH t |ts tdt|j n$tr.tdnt| t | j|¡S d S )Nz*repeat is not implemented when repeats is z)repeat is not implemented for pyarrow < 7) ry rñ r rL rÜ r r9 Z binary_repeatr~ )r r® r> r> r? Ú _str_repeatÓ s
|
ÿ
|
z ArrowExtensionArray._str_repeatz Scalar | None)r r r r c C s( | d¡sd| }| j||||ddS )Nú^T©r )Ú
|
startswithr ©r r r r r r> r> r? Ú
|
_str_matchÝ s
|
|
z ArrowExtensionArray._str_match)r r r c C s. | d¡r| d¡r | d}| ||||¡S )Nú$z//$)Úendswithr´ r³ r> r> r? Ú_str_fullmatchä s
|
z"ArrowExtensionArray._str_fullmatch)r] ÚstartÚendc C s¤ |dkrZ|d k rZt j| j||d}t ||¡}t |d¡}t ||| ¡}t |||¡}n>|dkr~|d kr~| j}t ||¡}ntd|d|d|t| |S )Nr ©Ústoprl z find not implemented with sub=z, start=z, end=) r9 Úutf8_slice_codeunitsr~ Zfind_substringr r[ r¥ r rL )r r] r¸ r¹ ZslicesrS Ú not_foundZ offset_resultr> r> r? Ú _str_findë s ÿz ArrowExtensionArray._str_find)Úic
|
C s¬ t | j¡}|dkr2t ||¡}|}|d }d}n t | |¡}|}|d }d}t | d¡¡}t j| j|||d}tj d g| j
|
¡ | jj d} t ||| ¡} t | | S )Nr rk rl T©r¸ r» Ústepr³ ) r9 Ú utf8_lengthr~ Ú greater_equalÚgreaterrÈ r. r¼ rJ r ræ rL r¥ )
|
r r¿ ÚlengthsZ out_of_boundsr¸ r» rÁ Znot_out_of_boundsÚselectedrS r> r> r? Ú_str_getû s( ÿ z ArrowExtensionArray._str_get©Úsepc C s t | t | j|¡S r8 )rL r9 Z binary_joinr~ )r rÉ r> r> r? Ú _str_join s z ArrowExtensionArray._str_join)rÉ Úexpandc s( fdd}| |¡}t| t |¡S )Nc s
|
| ¡S r8 )Ú partition©rà rÈ r> r? r@ rA z4ArrowExtensionArray._str_partition.<locals>.<lambda>©r rL rJ r ©r rÉ rË Ú predicaterS r> rÈ r? Ú_str_partition s
|
z"ArrowExtensionArray._str_partitionc s( fdd}| |¡}t| t |¡S )Nc s
|
| ¡S r8 )Ú
|
rpartitionrÍ rÈ r> r? r@ rA z5ArrowExtensionArray._str_rpartition.<locals>.<lambda>rÎ rÏ r> rÈ r? Ú_str_rpartition s
|
z#ArrowExtensionArray._str_rpartitionrÀ c C s4 |d kr d}|d krd}t | tj| j|||dS )Nr rk rÀ )rL r9 r¼ r~ )r r¸ r» rÁ r> r> r? Ú
|
_str_slice s ÿz ArrowExtensionArray._str_slicez
|
str | None)r¸ r» r¨ c C s2 |d kr d}|d krd}t | t | j|||¡S )Nrä r )rL r9 Zutf8_replace_slicer~ )r r¸ r» r¨ r> r> r? Ú_str_slice_replace' s
|
z&ArrowExtensionArray._str_slice_replacec C s t | t | j¡S r8 )rL r9 Z utf8_is_alnumr~ rÉ r> r> r? Ú _str_isalnum0 s z ArrowExtensionArray._str_isalnumc C s t | t | j¡S r8 )rL r9 Z utf8_is_alphar~ rÉ r> r> r? Ú _str_isalpha3 s z ArrowExtensionArray._str_isalphac C s t | t | j¡S r8 )rL r9 Zutf8_is_decimalr~ rÉ r> r> r? Ú_str_isdecimal6 s z"ArrowExtensionArray._str_isdecimalc C s t | t | j¡S r8 )rL r9 Z utf8_is_digitr~ rÉ r> r> r? Ú _str_isdigit9 s z ArrowExtensionArray._str_isdigitc C s t | t | j¡S r8 )rL r9 Z utf8_is_lowerr~ rÉ r> r> r? Ú _str_islower< s z ArrowExtensionArray._str_islowerc C s t | t | j¡S r8 )rL r9 Zutf8_is_numericr~ rÉ r> r> r? Ú_str_isnumeric? s z"ArrowExtensionArray._str_isnumericc C s t | t | j¡S r8 )rL r9 Z utf8_is_spacer~ rÉ r> r> r? Ú _str_isspaceB s z ArrowExtensionArray._str_isspacec C s t | t | j¡S r8 )rL r9 Z utf8_is_titler~ rÉ r> r> r? Ú _str_istitleE s z ArrowExtensionArray._str_istitlec C s t | t | j¡S r8 )rL r9 Zutf8_capitalizer~ rÉ r> r> r? Ú_str_capitalizeH s z#ArrowExtensionArray._str_capitalizec C s t | t | j¡S r8 )rL r9 Z
|
utf8_titler~ rÉ r> r> r? Ú
|
_str_titleK s z ArrowExtensionArray._str_titlec C s t | t | j¡S r8 )rL r9 Z utf8_is_upperr~ rÉ r> r> r? Ú _str_isupperN s z ArrowExtensionArray._str_isupperc C s t | t | j¡S r8 )rL r9 Z utf8_swapcaser~ rÉ r> r> r? Ú _str_swapcaseQ s z!ArrowExtensionArray._str_swapcasec C s t | t | j¡S r8 )rL r9 r r~ rÉ r> r> r? Ú_str_lenT s z ArrowExtensionArray._str_lenc C s t | t | j¡S r8 )rL r9 Z
|
utf8_lowerr~ rÉ r> r> r? Ú
|
_str_lowerW s z ArrowExtensionArray._str_lowerc C s t | t | j¡S r8 )rL r9 Z
|
utf8_upperr~ rÉ r> r> r? Ú
|
_str_upperZ s z ArrowExtensionArray._str_upperc C s2 |d krt | j¡}nt j| j|d}t| |S ©N)Ú
|
characters)r9 Zutf8_trim_whitespacer~ Z utf8_trimrL ©r Zto_striprS r> r> r? Ú
|
_str_strip] s z ArrowExtensionArray._str_stripc C s2 |d krt | j¡}nt j| j|d}t| |S rå )r9 Zutf8_ltrim_whitespacer~ Z
|
utf8_ltrimrL rç r> r> r? Ú _str_lstripd s z ArrowExtensionArray._str_lstripc C s2 |d krt | j¡}nt j| j|d}t| |S rå )r9 Zutf8_rtrim_whitespacer~ Z
|
utf8_rtrimrL rç r> r> r? Ú _str_rstripk s z ArrowExtensionArray._str_rstrip©Úprefixc sN t jdk r&ddlm} tj| d}n fdd}| |¡}t| t |¡S )N)é é r ©Ú removeprefixrë c s
|
| ¡S r8 rï rÍ rë r> r? r@ rA z7ArrowExtensionArray._str_removeprefix.<locals>.<lambda>)
|
ÚsysÚ version_infoZpandas.util._str_methodsrð Ú functoolsÚpartialr rL rJ r )r rì rð rÐ rS r> rë r? Ú_str_removeprefixr s
|
|
z%ArrowExtensionArray._str_removeprefix)Úsuffixc C sD t j| j|d}t j| jdt| d}t ||| j¡}t| |S )Nr¢ r rº )r9 r¦ r~ r¼ r¶ r¥ rL )r rö r¦ ÚremovedrS r> r> r? Ú_str_removesuffix s z%ArrowExtensionArray._str_removesuffixc C s$ dd }| |¡}t| t |¡S )Nc S s | ¡ S r8 )ÚcasefoldrÍ r> r> r? r@ rA z3ArrowExtensionArray._str_casefold.<locals>.<lambda>rÎ )r rÐ rS r> r> r? Ú _str_casefold s
|
z!ArrowExtensionArray._str_casefoldÚstrict©Úencodingr c s* fdd}| |¡}t| t |¡S )Nc s | ¡S r8 )r3 rÍ rü r> r? r@ rA z1ArrowExtensionArray._str_encode.<locals>.<lambda>rÎ )r rý r rÐ rS r> rü r? Ú _str_encode s
|
z ArrowExtensionArray._str_encode)r r rË c C s t dd S )Nz:str.extract not supported with pd.ArrowDtype(pa.string()).)r )r r r rË r> r> r? Ú _str_extract s ÿz ArrowExtensionArray._str_extractc s6 t j||d fdd}| |¡}t| t |¡S )N)r c s
|
| ¡S r8 )ÚfindallrÍ r± r> r? r@ rA z2ArrowExtensionArray._str_findall.<locals>.<lambda>)rª Úcompiler rL rJ r )r r r rÐ rS r> r± r? Ú _str_findall s
|
z ArrowExtensionArray._str_findallú|c C s¢ t | j|¡ ¡ }| ¡ ¡ }| tj |¡¡}g }|
|
¡ D ]F}|d kr^| dgt | ¡ q<t |t t|¡¡}| |
|
¡ ¡ q<t| t |¡}||
|
¡ fS rï )r9 Ú split_patternr~ rÑ ÚflattenrM r¸ rJ Úcomputer Ú to_pylistÚappendr¶ r/ r ÚsetrL ) r rÉ rr r; Zuniques_sortedZ result_dataÚlstr rS r> r> r? Ú_str_get_dummies s z$ArrowExtensionArray._str_get_dummiesc s, fdd}| |¡}t| t |¡S )Nc s | ¡S r8 )r rÍ ©r¹ r¸ r] r> r? r@ rA z0ArrowExtensionArray._str_index.<locals>.<lambda>rÎ ©r r] r¸ r¹ rÐ rS r> r r? Ú
|
_str_index¬ s
|
z ArrowExtensionArray._str_indexc s, fdd}| |¡}t| t |¡S )Nc s | ¡S r8 )ÚrindexrÍ r r> r? r@ ² rA z1ArrowExtensionArray._str_rindex.<locals>.<lambda>rÎ r r> r r? Ú _str_rindex± s
|
z ArrowExtensionArray._str_rindex©Úformc s( fdd}| |¡}t| t |¡S )Nc s t | ¡S r8 )Ú unicodedataÚ normalizerÍ r r> r? r@ · rA z4ArrowExtensionArray._str_normalize.<locals>.<lambda>rÎ )r r rÐ rS r> r r? Ú_str_normalize¶ s
|
z"ArrowExtensionArray._str_normalize)r] r¸ c s, fdd}| |¡}t| t |¡S )Nc s | ¡S r8 )ÚrfindrÍ r r> r? r@ ¼ rA z0ArrowExtensionArray._str_rfind.<locals>.<lambda>rÎ r r> r r? Ú
|
_str_rfind» s
|
z ArrowExtensionArray._str_rfindz bool | None)r rw rË r c C s6 |dkr d }|rt j}nt j}t| || j||dS )N¾ r rl )Ú
|
max_splits)r9 Zsplit_pattern_regexr rL r~ )r r rw rË r Z
|
split_funcr> r> r? Ú
|
_str_splitÀ s z ArrowExtensionArray._str_split)r rw c C s( |dkr d }t | tj| j||ddS )Nr T)r Úreverse)rL r9 r r~ )r r rw r> r> r? Ú _str_rsplitÏ s z ArrowExtensionArray._str_rsplitzdict[int, str]©Útablec s( fdd}| |¡}t| t |¡S )Nc s
|
| ¡S r8 )Ú translaterÍ r r> r? r@ Õ rA z4ArrowExtensionArray._str_translate.<locals>.<lambda>rÎ )r r rÐ rS r> r r? Ú_str_translateÔ s
|
z"ArrowExtensionArray._str_translate)r c s< ||d< t jf | fdd}| |¡}t| t |¡S )Nr c s d | ¡¡S )NÚ
|
)ÚjoinÚwraprÍ ©Ztwr> r? r@ Ü rA z/ArrowExtensionArray._str_wrap.<locals>.<lambda>)ÚtextwrapÚ TextWrapperr rL rJ r )r r r rÐ rS r> r$ r? Ú _str_wrapÙ s
|
|
z ArrowExtensionArray._str_wrapc C s t | t | j¡S r8 )rL r9 Úyearr~ rÉ r> r> r? Ú_dt_yearà s z ArrowExtensionArray._dt_yearc C s t | t | j¡S r8 )rL r9 Údayr~ rÉ r> r> r? Ú_dt_dayä s zArrowExtensionArray._dt_dayc C s t | t | j¡S r8 )rL r9 Z day_of_weekr~ rÉ r> r> r? Ú_dt_day_of_weekè s z#ArrowExtensionArray._dt_day_of_weekc C s t | t | j¡S r8 )rL r9 Z day_of_yearr~ rÉ r> r> r? Ú_dt_day_of_yearï s z#ArrowExtensionArray._dt_day_of_yearc C s t | t | j¡S r8 )rL r9 Úhourr~ rÉ r> r> r? Ú_dt_hourõ s z ArrowExtensionArray._dt_hourc C s t | t | j¡S r8 )rL r9 Z iso_calendarr~ rÉ r> r> r? Ú_dt_isocalendarù s z#ArrowExtensionArray._dt_isocalendarc C s t | t | j¡S r8 )rL r9 Z is_leap_yearr~ rÉ r> r> r? Ú_dt_is_leap_yearü s z$ArrowExtensionArray._dt_is_leap_yearc C s t | t | j¡S r8 )rL r9 Ú microsecondr~ rÉ r> r> r? Ú_dt_microsecond s z#ArrowExtensionArray._dt_microsecondc C s t | t | j¡S r8 )rL r9 Úminuter~ rÉ r> r> r? Ú
|
_dt_minute s z ArrowExtensionArray._dt_minutec C s t | t | j¡S r8 )rL r9 Úmonthr~ rÉ r> r> r? Ú _dt_month s z ArrowExtensionArray._dt_monthc C s t | t | j¡S r8 )rL r9 Ú
|
nanosecondr~ rÉ r> r> r? Ú_dt_nanosecond s z"ArrowExtensionArray._dt_nanosecondc C s t | t | j¡S r8 )rL r9 Úquarterr~ rÉ r> r> r? Ú _dt_quarter s z ArrowExtensionArray._dt_quarterc C s t | t | j¡S r8 )rL r9 Úsecondr~ rÉ r> r> r? Ú
|
_dt_second s z ArrowExtensionArray._dt_secondc C s t | | j t ¡ ¡S r8 )rL r~ r
|
rJ Zdate32rÉ r> r> r? Ú_dt_date s z ArrowExtensionArray._dt_datec C s6 | j jjdkr| j jjnd}t| | j t |¡¡S )N> rp ro rp )rx rz rt rL r~ r
|
rJ Ztime64)r rt r> r> r? Ú_dt_time s
|
ÿ ýz ArrowExtensionArray._dt_timec C s
|
| j jjS r8 )rx rz ÚtzrÉ r> r> r? Ú_dt_tz% s zArrowExtensionArray._dt_tz©Úformatc C s t | tj| j|dS )NrB )rL r9 Ústrftimer~ )r rC r> r> r? Ú _dt_strftime) s z ArrowExtensionArray._dt_strftimer z#Literal[('ceil', 'floor', 'round')]r r )r Ú ambiguousÚ nonexistentc
|
C s´ |dkrt d|dkr t dt|}|d kr>td| dddddddd d
|
d d d ddd}| |jd ¡}|d krtd|d|j}tt| d} t| | | j ||dS )Nr zambiguous is not supported.z nonexistent is not supported.z Must specify a valid frequency: r( r: r6 Úweekr* r. r4 r< Z millisecondr2 r8 )ÚAZASÚQZQSÚMZMSÚWÚDÚHÚTÚSÚLÚUÚNzfreq=ú is not supportedZ _temporal)Úmultiplert )
|
r r/ rs r Ú_prefixrw r r9 rL r~ )
|
r r ÚfreqrF rG ÚoffsetZpa_supported_unitrt rU Zrounding_methodr> r> r? Ú_round_temporally, s8 òz%ArrowExtensionArray._round_temporally©rF rG c C s | d|||¡S )NÚceil©rY ©r rW rF rG r> r> r? Ú_dt_ceilQ s z ArrowExtensionArray._dt_ceilc C s | d|||¡S )NrQ r\ r] r> r> r? Ú _dt_floorY s z ArrowExtensionArray._dt_floorc C s | d|||¡S )NrA r\ r] r> r> r? Ú _dt_rounda s z ArrowExtensionArray._dt_roundc C s4 | j ¡ }| jjjdkr&dd |D }tj|tdS )Nrp c S s$ g | ] }|d krd n
|
|j ddqS )NF)Úwarn)Z to_pydatetime)rU Útsr> r> r? rV l s z9ArrowExtensionArray._dt_to_pydatetime.<locals>.<listcomp>r| )r~ r r rz rt r´ r rD )r r r> r> r? Ú_dt_to_pydatetimei s
|
z%ArrowExtensionArray._dt_to_pydatetimec C s |dkrt d|ddddd |d ¡}|d krDt d|d|d krf| j t | jjj¡¡}nt j
|
| jt |||d}t | |S ) Nr z
|
ambiguous=rT ZearliestZlatest)r Zshift_backwardZ shift_forwardz nonexistent=rZ ) r r r~ r
|
rJ Ú timestamprx rz rt r9 Zassume_timezonerq rL )r r@ rF rG Znonexistent_parS r> r> r? Ú_dt_tz_localizeo s( ý û ÿz#ArrowExtensionArray._dt_tz_localize)N)N)T)T)NNN)T)r )rO N)FN)T)T)r )rO r )Tr NT)N)N)rl Tr T)Tr N)Tr N)r N)NNN)NNN)N)N)N)rû )r T)r )r )r N)r N)r N)Nrl FN)Nrl )r r )r r )r r )r r )r r )rÜ Ú
|
__module__Ú __qualname__Ú__doc__Ú__annotations__r Ú classmethodr r© rÁ rÄ rÅ rÇ rÊ rË rÍ rÎ rÔ rÖ rã rê rì rî rð Úpropertyrx rò ró rù rý r& rö r r r r r! r r# r r) r r0 r1 r< r> rA rE r¸ rª r rI rM rQ rX rY r rx r r r rr rt ru r r r r r¥ r§ r r¯ r´ r· r¾ rÇ rÊ rÑ rÓ rÔ rÕ rÖ r× rØ rÙ rÚ rÛ rÜ rÝ rÞ rß rà rá râ rã rä rè ré rê rõ rø rú rþ rÿ r r r r r r r r r r' r) r+ r, Z _dt_dayofweekZ _dt_weekdayr- Z _dt_dayofyearr/ r0 r1 r3 r5 r7 r9 r; r= r> r? rA rE rY r^ r_ r` rc re Ú __classcell__r> r> rû r? rh µ s´
|
+ ÿ>E 1
|
:=û
|
ü 9
|
þ ÿ ü übü&/ÿ2xRù C')+4
|
ü ÿ
|
ù ÿ ÿ ÿ ÿ
|
û û ( ü ü ü ü)nÚ
|
__future__r r r ró rå rª rñ r% Útypingr r r r r r r
|
r Únumpyr´ Z pandas._libsr Zpandas._typingr r r r r r r r r r r r r Z pandas.compatr r r r Zpandas.util._decoratorsr Zpandas.util._validatorsr Zpandas.core.dtypes.commonr r r! r" r# r$ r% Zpandas.core.dtypes.missingr& Z pandas.corer' Zpandas.core.arrayliker( Zpandas.core.arrays.baser) r* Zpandas.core.commonÚcoreÚcommonrs Zpandas.core.indexersr+ r, r- Zpandas.core.strings.baser. Zpandas.tseries.frequenciesr/ r² rJ Zpyarrow.computer r9 Z%pandas.core.arrays.arrow._arrow_utilsr0 Z pandas.core.arrays.arrow.dtyper1 r rk ZlessrÄ Z
|
less_equalrà rÛ r: rB rC rè rN rT rU rV rW rç rZ rí rd re rP rf rg rv r} rh r> r> r> r? Ú<module> s $ < $ ú
|
ú ð
|
|
ÿ
|
