Sindbad~EG File Manager
�
Mٜg7L � �4 � U d Z ddlmZ ddlZddlmZmZmZmZm Z m
Z
m Z ddl Z ddl
ZddlmZmZ ddlmZ ddlmZ ddlmZ dd lmZm Z m Z dd
l m Z dd l m!Z!m"Z"m#Z# dd l$m%Z% dd
l&m'Z'm(Z(m)Z) ddl*m+Z+ ddl,m-Z-m.Z.m/Z/ ddl0m1Z1 ddl2m3Z3m4Z4m5Z5 ddl6m7Z7m8Z8m9Z9m:Z:m;Z;m<Z<m=Z= ddl>m?Z? ddl@mAZA ddlBmCZCmDZD er4ddlEmFZFmGZG ddlHmIZImJZJmKZKmLZLmMZMmNZNmOZOmPZPmQZQmRZRmSZSmTZTmUZUmVZVmWZWmXZXmYZY ddlZm[Z[ i Z\de]d< G d� d � Z^ G d � d e^� Z_ G d � d � Z` G d!� d"e`� Zay)#z�
An interface for extending pandas with custom arrays.
.. warning::
This is an experimental API and subject to breaking changes
without warning.
� )� annotationsN)�
TYPE_CHECKING�Any�Callable�ClassVar�Literal�cast�overload)�algos�lib)�set_function_name)�function��AbstractMethodError)�Appender� Substitution�cache_readonly)�find_stack_level)�validate_bool_kwarg�validate_fillna_kwargs�validate_insert_loc)�maybe_cast_pointwise_result)� is_list_like� is_scalar� pandas_dtype)�ExtensionDtype)� ABCDataFrame�ABCIndex� ABCSeries��isna)� arraylike�missing� roperator)�
duplicated�factorize_array�isin� map_array�mode�rank�unique)�quantile_with_mask)�_fill_limit_area_1d)�
nargminmax�nargsort)�Iterator�Sequence)� ArrayLike� AstypeArg�AxisInt�Dtype�DtypeObj�
FillnaOptions�InterpolateOptions� NumpySorter�NumpyValueArrayLike�PositionalIndexer�
ScalarIndexer�Self�SequenceIndexer�Shape�SortKind� TakeIndexer�npt)�Indexzdict[str, str]� _extension_array_shared_docsc �� � e Zd ZU dZdZdZeddd�d`d�� Zedad�� Zeddd� d`d �� Z ed
� � Z
e dbd �� Z e dcd �� Z ddd
�Z ded�Z
dfd�Zdgd�Zdhd�Zdid�Zdid�Zddej( f djd�Zedkd�� Zedld�� Zedfd�� Zedfd�� Zedfd�� Ze dmdnd�� Z e dmdod�� Z e dmdpd �� Z dqdpd �Z drd �Z edsd �� Z dtd!�Z d d"d#d$� dud%�Z dqdvd&�Z!dqdvd'�Z" dwd(�Z#ddd d)� dxd*�Z$ dy dzd+�Z%d{d,�Z& d| d}d.�Z'd~dd/�Z(d{d0�Z) d� d�d1�Z*d�d2�Z+d�d3�Z,d�d4�Z- dq d�d5�Z.d6e/d7< e0d �8� e1e/d7 � d�d�d9�� � Z2ddd:� d�d;�Z3d{d<�Z4d�d�d=�Z5d�d>�Z6d�d?�Z7d�d@�Z8d�d�dA�Z9d�dB�Z:ed�dC�� Z;d�d�dD�Z<ed�dE�� Z=e>dsdF�� Z?d dG� d�dH�Z@d ddI� d�dJ�ZAdKeBdL< dtdM�ZC d�dN�ZDd�dO�ZEd�dP�ZFd�dQ�ZGd�dR�ZHd�dS�ZId�dT�ZJ d�dU�ZKdVdWd-d ddX� d�dY�ZLed�dZ�� ZMd�d[�ZNdqd�d\�ZOd�d]�ZPd�d^�ZQ d�d_�ZRy)��ExtensionArrayav
Abstract base class for custom 1-D array types.
pandas will recognize instances of this class as proper arrays
with a custom type and will not attempt to coerce them to objects. They
may be stored directly inside a :class:`DataFrame` or :class:`Series`.
Attributes
----------
dtype
nbytes
ndim
shape
Methods
-------
argsort
astype
copy
dropna
duplicated
factorize
fillna
equals
insert
interpolate
isin
isna
ravel
repeat
searchsorted
shift
take
tolist
unique
view
_accumulate
_concat_same_type
_explode
_formatter
_from_factorized
_from_sequence
_from_sequence_of_strings
_hash_pandas_object
_pad_or_backfill
_reduce
_values_for_argsort
_values_for_factorize
Notes
-----
The interface includes the following abstract methods that must be
implemented by subclasses:
* _from_sequence
* _from_factorized
* __getitem__
* __len__
* __eq__
* dtype
* nbytes
* isna
* take
* copy
* _concat_same_type
* interpolate
A default repr displaying the type, (truncated) data, length,
and dtype is provided. It can be customized or replaced by
by overriding:
* __repr__ : A default repr for the ExtensionArray.
* _formatter : Print scalars inside a Series or DataFrame.
Some methods require casting the ExtensionArray to an ndarray of Python
objects with ``self.astype(object)``, which may be expensive. When
performance is a concern, we highly recommend overriding the following
methods:
* fillna
* _pad_or_backfill
* dropna
* unique
* factorize / _values_for_factorize
* argsort, argmax, argmin / _values_for_argsort
* searchsorted
* map
The remaining methods implemented on this class should be performant,
as they only compose abstract methods. Still, a more efficient
implementation may be available, and these methods can be overridden.
One can implement methods to handle array accumulations or reductions.
* _accumulate
* _reduce
One can implement methods to handle parsing from strings that will be used
in methods such as ``pandas.io.parsers.read_csv``.
* _from_sequence_of_strings
This class does not inherit from 'abc.ABCMeta' for performance reasons.
Methods and properties required by the interface raise
``pandas.errors.AbstractMethodError`` and no ``register`` method is
provided for registering virtual subclasses.
ExtensionArrays are limited to 1 dimension.
They may be backed by none, one, or many NumPy arrays. For example,
``pandas.Categorical`` is an extension array backed by two arrays,
one for codes and one for categories. An array of IPv6 address may
be backed by a NumPy structured array with two fields, one for the
lower 64 bits and one for the upper 64 bits. Or they may be backed
by some other storage type, like Python lists. Pandas makes no
assumptions on how the data are stored, just that it can be converted
to a NumPy array.
The ExtensionArray interface does not impose any rules on how this data
is stored. However, currently, the backing data cannot be stored in
attributes called ``.values`` or ``._values`` to ensure full compatibility
with pandas internals. But other names as ``.data``, ``._data``,
``._items``, ... can be freely used.
If implementing NumPy's ``__array_ufunc__`` interface, pandas expects
that
1. You defer by returning ``NotImplemented`` when any Series are present
in `inputs`. Pandas will extract the arrays and call the ufunc again.
2. You define a ``_HANDLED_TYPES`` tuple as an attribute on the class.
Pandas inspect this to determine whether the ufunc is valid for the
types present.
See :ref:`extending.extension.ufunc` for more.
By default, ExtensionArrays are not hashable. Immutable subclasses may
override this behavior.
Examples
--------
Please see the following:
https://github.com/pandas-dev/pandas/blob/main/pandas/tests/extension/list/array.py
� extensioni� NF��dtype�copyc � � t | � �)a�
Construct a new ExtensionArray from a sequence of scalars.
Parameters
----------
scalars : Sequence
Each element will be an instance of the scalar type for this
array, ``cls.dtype.type`` or be converted into this type in this method.
dtype : dtype, optional
Construct for this particular dtype. This should be a Dtype
compatible with the ExtensionArray.
copy : bool, default False
If True, copy the underlying data.
Returns
-------
ExtensionArray
Examples
--------
>>> pd.arrays.IntegerArray._from_sequence([4, 5])
<IntegerArray>
[4, 5]
Length: 2, dtype: Int64
r )�cls�scalarsrI rJ s �B/usr/local/lib/python3.12/site-packages/pandas/core/arrays/base.py�_from_sequencez ExtensionArray._from_sequence
� � �6 "�#�&�&� c � � | j ||d�� S # t t f$ r � t $ r! t j
dt
� �� � w xY w)a�
Strict analogue to _from_sequence, allowing only sequences of scalars
that should be specifically inferred to the given dtype.
Parameters
----------
scalars : sequence
dtype : ExtensionDtype
Raises
------
TypeError or ValueError
Notes
-----
This is called in a try/except block when casting the result of a
pointwise operation.
FrH zm_from_scalars should only raise ValueError or TypeError. Consider overriding _from_scalars where appropriate.��
stacklevel)rO �
ValueError� TypeError� Exception�warnings�warnr )rL rM rI s rN �
_from_scalarsz ExtensionArray._from_scalars* s[ � �(
��%�%�g�U��%�G� G���I�&� � �� � �M�M�G�+�-�
�
�
�s � �9Ac � � t | � �)a�
Construct a new ExtensionArray from a sequence of strings.
Parameters
----------
strings : Sequence
Each element will be an instance of the scalar type for this
array, ``cls.dtype.type``.
dtype : dtype, optional
Construct for this particular dtype. This should be a Dtype
compatible with the ExtensionArray.
copy : bool, default False
If True, copy the underlying data.
Returns
-------
ExtensionArray
Examples
--------
>>> pd.arrays.IntegerArray._from_sequence_of_strings(["1", "2", "3"])
<IntegerArray>
[1, 2, 3]
Length: 3, dtype: Int64
r )rL �stringsrI rJ s rN �_from_sequence_of_stringsz(ExtensionArray._from_sequence_of_stringsJ s � �: "�#�&�&rQ c � � t | � �)a�
Reconstruct an ExtensionArray after factorization.
Parameters
----------
values : ndarray
An integer ndarray with the factorized values.
original : ExtensionArray
The original ExtensionArray that factorize was called on.
See Also
--------
factorize : Top-level factorize method that dispatches here.
ExtensionArray.factorize : Encode the extension array as an enumerated type.
Examples
--------
>>> interv_arr = pd.arrays.IntervalArray([pd.Interval(0, 1),
... pd.Interval(1, 5), pd.Interval(1, 5)])
>>> codes, uniques = pd.factorize(interv_arr)
>>> pd.arrays.IntervalArray._from_factorized(uniques, interv_arr)
<IntervalArray>
[(0, 1], (1, 5]]
Length: 2, dtype: interval[int64, right]
r )rL �values�originals rN �_from_factorizedz ExtensionArray._from_factorizedi rP rQ c � � y �N� ��self�items rN � __getitem__zExtensionArray.__getitem__� � � � rQ c � � y rc rd re s rN rh zExtensionArray.__getitem__� ri rQ c � � t | � �)an
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'
* list[int]: A list of int
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 re s rN rh zExtensionArray.__getitem__� s � �> "�$�'�'rQ c �0 � t t | � � d�� �)a^
Set one or more values inplace.
This method is not required to satisfy the pandas extension array
interface.
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
z does not implement __setitem__.)�NotImplementedError�type)rf �key�values rN � __setitem__zExtensionArray.__setitem__� s � �V "�T�$�Z�L�0P�"Q�R�RrQ c � � t | � �)z\
Length of this array
Returns
-------
length : int
r �rf s rN �__len__zExtensionArray.__len__� s � � "�$�'�'rQ c # �L K � t t | � � D ] }| | �� � y�w)z5
Iterate over elements of the array.
N)�range�len)rf �is rN �__iter__zExtensionArray.__iter__� s$ � �� � �s�4�y�!�A��q�'�M� "�s �"$c � � t |� r]t |� rR| j sy|| j j u s t || j j � r | j S y|| k( j � S )z,
Return for `item in self`.
F) r r! � _can_hold_narI �na_value�
isinstancern �_hasna�anyre s rN � __contains__zExtensionArray.__contains__� sb � � �T�?�t�D�z��$�$� �����,�,�,�
�4���� � �0Q��{�{�"� � �D�L�%�%�'� 'rQ c � � t | � �)zE
Return for `self == other` (element-wise equality).
r �rf �others rN �__eq__zExtensionArray.__eq__ � � � "�$�'�'rQ c � � | |k( S )zH
Return for `self != other` (element-wise in-equality).
rd r� s rN �__ne__zExtensionArray.__ne__ s � �
�� � � rQ c �� � t j | |�� }|s|t j ur|j � }|t j ur||| j � < |S )aI
Convert to a NumPy ndarray.
This is similar to :meth:`numpy.asarray`, but may provide additional control
over how the conversion is done.
Parameters
----------
dtype : str or numpy.dtype, optional
The dtype to pass to :meth:`numpy.asarray`.
copy : bool, default False
Whether to ensure that the returned value is a not a view on
another array. Note that ``copy=False`` does not *ensure* that
``to_numpy()`` is no-copy. Rather, ``copy=True`` ensure that
a copy is made, even if not strictly necessary.
na_value : Any, optional
The value to use for missing values. The default value depends
on `dtype` and the type of the array.
Returns
-------
numpy.ndarray
�rI )�np�asarrayr �
no_defaultrJ r! )rf rI rJ r| �results rN �to_numpyzExtensionArray.to_numpy sN � �: ���D��.�� �8�3�>�>�1��[�[�]�F� �3�>�>� )�"*�F�4�9�9�;� ��
rQ c � � t | � �)z�
An instance of ExtensionDtype.
Examples
--------
>>> pd.array([1, 2, 3]).dtype
Int64Dtype()
r rs s rN rI zExtensionArray.dtypeC r� rQ c � � t | � fS )z�
Return a tuple of the array dimensions.
Examples
--------
>>> arr = pd.array([1, 2, 3])
>>> arr.shape
(3,)
)rw rs s rN �shapezExtensionArray.shapeO s � � �D� �|�rQ c �@ � t j | j � S )z6
The number of elements in the array.
)r� �prodr� rs s rN �sizezExtensionArray.size\ s � � �w�w�t�z�z�"�"rQ c � � y)z�
Extension Arrays are only allowed to be 1-dimensional.
Examples
--------
>>> arr = pd.array([1, 2, 3])
>>> arr.ndim
1
� rd rs s rN �ndimzExtensionArray.ndime s � � rQ c � � t | � �)z�
The number of bytes needed to store this object in memory.
Examples
--------
>>> pd.array([1, 2, 3]).nbytes
27
r rs s rN �nbyteszExtensionArray.nbytesr s � � "�$�'�'rQ c � � y rc rd �rf rI rJ s rN �astypezExtensionArray.astype� ri rQ c � � y rc rd r� s rN r� zExtensionArray.astype� ri rQ c � � y rc rd r� s rN r� zExtensionArray.astype� ri rQ Tc �� � t |� }|| j k( r|s| S | j � S t |t � r$|j � }|j
| ||�� S t j |d� rddl m
} |j
| ||�� S t j |d� rddl m } |j
| ||�� S |st j | |�� S t j | ||�� S )a�
Cast to a NumPy array or ExtensionArray with 'dtype'.
Parameters
----------
dtype : str or dtype
Typecode or data-type to which the array is cast.
copy : bool, default True
Whether to copy the data, even if not necessary. If False,
a copy is made only if the old dtype does not match the
new dtype.
Returns
-------
np.ndarray or pandas.api.extensions.ExtensionArray
An ``ExtensionArray`` if ``dtype`` is ``ExtensionDtype``,
otherwise a Numpy ndarray with ``dtype`` for its dtype.
Examples
--------
>>> arr = pd.array([1, 2, 3])
>>> arr
<IntegerArray>
[1, 2, 3]
Length: 3, dtype: Int64
Casting to another ``ExtensionDtype`` returns an ``ExtensionArray``:
>>> arr1 = arr.astype('Float64')
>>> arr1
<FloatingArray>
[1.0, 2.0, 3.0]
Length: 3, dtype: Float64
>>> arr1.dtype
Float64Dtype()
Otherwise, we will get a Numpy ndarray:
>>> arr2 = arr.astype('float64')
>>> arr2
array([1., 2., 3.])
>>> arr2.dtype
dtype('float64')
rH �Mr )�
DatetimeArray�m)�TimedeltaArrayr� )r rI rJ r} r �construct_array_typerO r � is_np_dtype�pandas.core.arraysr� r� r� r� �array)rf rI rJ rL r� r� s rN r� zExtensionArray.astype� s� � �Z �U�#�� �D�J�J� ��� ��y�y�{�"� �e�^� ,��,�,�.�C��%�%�d�%�d�%�C� C�
�_�_�U�C�
(� 8� �/�/��E��/�M� M�
�_�_�U�C�
(� 9�!�0�0��U��0�N� N���:�:�d�%�0� 0��8�8�D��D�9� 9rQ c � � t | � �)a'
A 1-D array indicating if each value is missing.
Returns
-------
numpy.ndarray or pandas.api.extensions.ExtensionArray
In most cases, this should return a NumPy ndarray. For
exceptional cases like ``SparseArray``, where returning
an ndarray would be expensive, an ExtensionArray may be
returned.
Notes
-----
If returning an ExtensionArray, then
* ``na_values._is_boolean`` should be True
* `na_values` should implement :func:`ExtensionArray._reduce`
* ``na_values.any`` and ``na_values.all`` should be implemented
Examples
--------
>>> arr = pd.array([1, 2, np.nan, np.nan])
>>> arr.isna()
array([False, False, True, True])
r rs s rN r! zExtensionArray.isna� s � �4 "�$�'�'rQ c �P � t | j � j � � S )z�
Equivalent to `self.isna().any()`.
Some ExtensionArray subclasses may be able to optimize this check.
)�boolr! r rs s rN r~ zExtensionArray._hasna� s � � �D�I�I�K�O�O�%�&�&rQ c �, � t j | � S )a�
Return values for sorting.
Returns
-------
ndarray
The transformed values should maintain the ordering between values
within the array.
See Also
--------
ExtensionArray.argsort : Return the indices that would sort this array.
Notes
-----
The caller is responsible for *not* modifying these values in-place, so
it is safe for implementers to give views on ``self``.
Functions that use this (e.g. ``ExtensionArray.argsort``) should ignore
entries with missing values in the original array (according to
``self.isna()``). This means that the corresponding entries in the returned
array don't need to be modified to sort correctly.
Examples
--------
In most cases, this is the underlying Numpy array of the ``ExtensionArray``:
>>> arr = pd.array([1, 2, 3])
>>> arr._values_for_argsort()
array([1, 2, 3])
)r� r� rs s rN �_values_for_argsortz"ExtensionArray._values_for_argsort� s � �B �x�x��~� rQ � quicksort�last)� ascending�kind� na_positionc
� � t j |d|� }| j � }t ||||t j
| j
� � �� S )a�
Return the indices that would sort this array.
Parameters
----------
ascending : bool, default True
Whether the indices should result in an ascending
or descending sort.
kind : {'quicksort', 'mergesort', 'heapsort', 'stable'}, optional
Sorting algorithm.
na_position : {'first', 'last'}, default 'last'
If ``'first'``, put ``NaN`` values at the beginning.
If ``'last'``, put ``NaN`` values at the end.
*args, **kwargs:
Passed through to :func:`numpy.argsort`.
Returns
-------
np.ndarray[np.intp]
Array of indices that sort ``self``. If NaN values are contained,
NaN values are placed at the end.
See Also
--------
numpy.argsort : Sorting implementation used internally.
Examples
--------
>>> arr = pd.array([3, 1, 2, 5, 4])
>>> arr.argsort()
array([1, 2, 0, 4, 3])
rd )r� r� r� �mask)�nv� validate_argsort_with_ascendingr� r/ r� r� r! )rf r� r� r� �kwargsr_ s rN �argsortzExtensionArray.argsort sR � �Z �6�6�y�"�f�M� ��)�)�+��� �� �#����D�I�I�K�(�
�
rQ c �Z � t |d� |s| j rt �t | d� S )a
Return the index of minimum value.
In case of multiple occurrences of the minimum value, the index
corresponding to the first occurrence is returned.
Parameters
----------
skipna : bool, default True
Returns
-------
int
See Also
--------
ExtensionArray.argmax : Return the index of the maximum value.
Examples
--------
>>> arr = pd.array([3, 1, 2, 5, 4])
>>> arr.argmin()
1
�skipna�argmin�r r~ rm r. �rf r� s rN r� zExtensionArray.argminX �* � �: �F�H�-��$�+�+�%� %��$��)�)rQ c �Z � t |d� |s| j rt �t | d� S )a
Return the index of maximum value.
In case of multiple occurrences of the maximum value, the index
corresponding to the first occurrence is returned.
Parameters
----------
skipna : bool, default True
Returns
-------
int
See Also
--------
ExtensionArray.argmin : Return the index of the minimum value.
Examples
--------
>>> arr = pd.array([3, 1, 2, 5, 4])
>>> arr.argmax()
3
r� �argmaxr� r� s rN r� zExtensionArray.argmaxz r� rQ c �D � t t | � j � d�� �)a�
See DataFrame.interpolate.__doc__.
Examples
--------
>>> arr = pd.arrays.NumpyExtensionArray(np.array([0, 1, np.nan, 3]))
>>> arr.interpolate(method="linear",
... limit=3,
... limit_direction="forward",
... index=pd.Index([1, 2, 3, 4]),
... fill_value=1,
... copy=False,
... axis=0,
... limit_area="inside"
... )
<NumpyExtensionArray>
[0.0, 1.0, 2.0, 3.0]
Length: 4, dtype: float64
z does not implement interpolate)rm rn �__name__) rf �method�axis�index�limit�limit_direction�
limit_arearJ r� s rN � interpolatezExtensionArray.interpolate� s( � �@ "��D�z�"�"�#�#B� C�
�
rQ )r� r� rJ c � � t | � j t j urt | � j t j u rZt j
dt t � �� |�!t t | � j � d�� �| j ||�� S | j � }|j � r�t j |� }t j |� }|� |j! � s t# ||� |dk( r*t% j& ||�� }| j) |d� � S t% j& |ddd
� |�� ddd
� }| ddd
� j) |d� � S |s| S | j+ � } | S ) a&
Pad or backfill values, used by Series/DataFrame ffill and bfill.
Parameters
----------
method : {'backfill', 'bfill', 'pad', 'ffill'}
Method to use for filling holes in reindexed Series:
* pad / ffill: propagate last valid observation forward to next valid.
* backfill / bfill: use NEXT valid observation to fill gap.
limit : int, default None
This is the maximum number of consecutive
NaN values to forward/backward fill. In other words, if there is
a gap with more than this number of consecutive NaNs, it will only
be partially filled. If method is not specified, this is the
maximum number of entries along the entire axis where NaNs will be
filled.
copy : bool, default True
Whether to make a copy of the data before filling. If False, then
the original should be modified and no new memory should be allocated.
For ExtensionArray subclasses that cannot do this, it is at the
author's discretion whether to ignore "copy=False" or to raise.
The base class implementation ignores the keyword if any NAs are
present.
Returns
-------
Same type as self
Examples
--------
>>> arr = pd.array([np.nan, np.nan, 2, 3, np.nan, np.nan])
>>> arr._pad_or_backfill(method="backfill", limit=1)
<IntegerArray>
[<NA>, 2, 2, 3, <NA>, <NA>]
Length: 6, dtype: Int64
z�ExtensionArray.fillna 'method' keyword is deprecated. In a future version. arr._pad_or_backfill will be called instead. 3rd-party ExtensionArray authors need to implement _pad_or_backfill.rS Nz� does not implement limit_area (added in pandas 2.2). 3rd-party ExtnsionArray authors need to add this argument to _pad_or_backfill.)r� r� �pad�r� T��
allow_fill���)rn �fillnarF �_pad_or_backfillrX rY �DeprecationWarningr rm r� r! r r# �clean_fill_methodr� r� �allr- �libalgos�get_fill_indexer�takerJ )
rf r� r� r� rJ r� �meth�npmask�indexer�
new_valuess
rN r� z ExtensionArray._pad_or_backfill� sk � �f
��J� � �^�%:�%:� :��T�
�+�+�~�/N�/N�N�
�M�M�$� #�+�-�
� �%�)��D�z�*�*�+� ,E� E�� �
�;�;�f�E�;�:� :��y�y�{�� �8�8�:��,�,�V�4�D��Z�Z��%�F��%�f�j�j�l�#�F�J�7��u�}�"�3�3�F�%�H���y�y��T�y�:�:� #�3�3�F�4�R�4�L��N�t�QS�t�T���D�b�D�z���w�4��@�@� �� � ����J��rQ c � � |�;t j dt | � j � d�t t � �� t
||� \ }}| j � }t j ||t | � � }|j � r�|��t j |� }t j |� }|dk( r*t j ||�� }| j# |d�� S t j |ddd � |�� ddd � }| ddd � j# |d�� S |s| dd } n| j% � } || |< | S |s| dd } | S | j% � } | S )
at
Fill NA/NaN values using the specified method.
Parameters
----------
value : scalar, array-like
If a scalar value is passed it is used to fill all missing values.
Alternatively, an array-like "value" can be given. It's expected
that the array-like have the same length as 'self'.
method : {'backfill', 'bfill', 'pad', 'ffill', None}, default None
Method to use for filling holes in reindexed Series:
* pad / ffill: propagate last valid observation forward to next valid.
* backfill / bfill: use NEXT valid observation to fill gap.
.. deprecated:: 2.1.0
limit : int, default None
If method is specified, this is the maximum number of consecutive
NaN values to forward/backward fill. In other words, if there is
a gap with more than this number of consecutive NaNs, it will only
be partially filled. If method is not specified, this is the
maximum number of entries along the entire axis where NaNs will be
filled.
.. deprecated:: 2.1.0
copy : bool, default True
Whether to make a copy of the data before filling. If False, then
the original should be modified and no new memory should be allocated.
For ExtensionArray subclasses that cannot do this, it is at the
author's discretion whether to ignore "copy=False" or to raise.
The base class implementation ignores the keyword in pad/backfill
cases.
Returns
-------
ExtensionArray
With NA/NaN filled.
Examples
--------
>>> arr = pd.array([np.nan, np.nan, 2, 3, np.nan, np.nan])
>>> arr.fillna(0)
<IntegerArray>
[0, 0, 2, 3, 0, 0]
Length: 6, dtype: Int64
NzThe 'method' keyword in z>.fillna is deprecated and will be removed in a future version.rS r� r� Tr� r� )rX rY rn r� �
FutureWarningr r r! r# �check_value_sizerw r r� r� r� r� r� r� rJ )
rf rp r� r� rJ r� r� r� r� r� s
rN r� zExtensionArray.fillna sk � �n � � �M�M�*�4��:�+>�+>�*?� @F� F� �+�-�
� /�u�f�=�
��v��y�y�{�� �(�(� �4��T��
�� �8�8�:��!� �0�0��8�����D�)���5�=� &� 7� 7��e� L�G� �9�9�W��9�>�>� '� 7� 7��t��t� �E� R�SW�UW�SW� X�G� ��"��:�?�?�7�t�?�D�D� �!%�a��J�!%����J�#(�
�4� � �� � !�!�W�
� �� "�Y�Y�[�
��rQ c �* � | | j � S )z�
Return ExtensionArray without NA values.
Returns
-------
Examples
--------
>>> pd.array([1, 2, np.nan]).dropna()
<IntegerArray>
[1, 2]
Length: 2, dtype: Int64
r rs s rN �dropnazExtensionArray.dropna� s � � �T�Y�Y�[�L�!�!rQ �keepc �| � | j � j t j d�� }t | ||�� S )aU
Return boolean ndarray denoting duplicate values.
Parameters
----------
keep : {'first', 'last', False}, default 'first'
- ``first`` : Mark duplicates as ``True`` except for the first occurrence.
- ``last`` : Mark duplicates as ``True`` except for the last occurrence.
- False : Mark all duplicates as ``True``.
Returns
-------
ndarray[bool]
Examples
--------
>>> pd.array([1, 1, 2, 3, 3], dtype="Int64").duplicated()
array([False, True, False, False, True])
F)rJ )r_ r� r� )r! r� r� �bool_r% )rf r� r� s rN r% zExtensionArray.duplicated� s2 � �, �y�y�{�!�!�"�(�(��!�7����D�t�<�<rQ c �` � t | � r|dk( r| j � S t |� r| j j }| j |gt
t |� t | � � z | j �� }|dkD r |}| d| }n| t |� d }|}| j ||g� S )a@
Shift values by desired number.
Newly introduced missing values are filled with
``self.dtype.na_value``.
Parameters
----------
periods : int, default 1
The number of periods to shift. Negative values are allowed
for shifting backwards.
fill_value : object, optional
The scalar value to use for newly introduced missing values.
The default is ``self.dtype.na_value``.
Returns
-------
ExtensionArray
Shifted.
Notes
-----
If ``self`` is empty or ``periods`` is 0, a copy of ``self`` is
returned.
If ``periods > len(self)``, then an array of size
len(self) is returned, with all values filled with
``self.dtype.na_value``.
For 2-dimensional ExtensionArrays, we are always shifting along axis=0.
Examples
--------
>>> arr = pd.array([1, 2, 3])
>>> arr.shift(2)
<IntegerArray>
[<NA>, <NA>, 1]
Length: 3, dtype: Int64
r r� N) rw rJ r! rI r| rO �min�abs�_concat_same_type)rf �periods�
fill_value�empty�a�bs rN �shiftzExtensionArray.shift� s� � �V �4�y�G�q�L��9�9�;� � �
� � ���,�,�J��#�#�
�L�3�s�7�|�S��Y�7� 7�t�z�z� $�
�� �Q�;��A��Y�w�h��A��S��\�^�$�A��A��%�%�q�!�f�-�-rQ c �x � t | j t � � }| j || j �� S )a@
Compute the ExtensionArray of unique values.
Returns
-------
pandas.api.extensions.ExtensionArray
Examples
--------
>>> arr = pd.array([1, 2, 3, 1, 2, 3])
>>> arr.unique()
<IntegerArray>
[1, 2, 3]
Length: 3, dtype: Int64
r� )r+ r� �objectrO rI )rf �uniquess rN r+ zExtensionArray.unique� s1 � � ����V�,�-���"�"�7�$�*�*�"�=�=rQ c � � | j t � }t |t � r|j t � }|j |||�� S )a8
Find indices where elements should be inserted to maintain order.
Find the indices into a sorted array `self` (a) such that, if the
corresponding elements in `value` were inserted before the indices,
the order of `self` would be preserved.
Assuming that `self` is sorted:
====== ================================
`side` returned index `i` satisfies
====== ================================
left ``self[i-1] < value <= self[i]``
right ``self[i-1] <= value < self[i]``
====== ================================
Parameters
----------
value : array-like, list or scalar
Value(s) to insert into `self`.
side : {'left', 'right'}, optional
If 'left', the index of the first suitable location found is given.
If 'right', return the last such index. If there is no suitable
index, return either 0 or N (where N is the length of `self`).
sorter : 1-D array-like, optional
Optional array of integer indices that sort array a into ascending
order. They are typically the result of argsort.
Returns
-------
array of ints or int
If value is array-like, array of insertion points.
If value is scalar, a single integer.
See Also
--------
numpy.searchsorted : Similar method from NumPy.
Examples
--------
>>> arr = pd.array([1, 2, 3, 5])
>>> arr.searchsorted([4])
array([3])
)�side�sorter)r� r� r} rF � searchsorted)rf rp r� r� �arrs rN r� zExtensionArray.searchsorted� sC � �n �k�k�&�!�� �e�^� ,��L�L��(�E�� � ��D�� �@�@rQ c �| � t | � t |� k7 ryt t |� }| j |j k7 ryt | � t |� k7 ry| |k( }t |t � r|j
d� }| j � |j � z }t ||z j � � S )a�
Return if another array is equivalent to this array.
Equivalent means that both arrays have the same shape and dtype, and
all values compare equal. Missing values in the same location are
considered equal (in contrast with normal equality).
Parameters
----------
other : ExtensionArray
Array to compare to this Array.
Returns
-------
boolean
Whether the arrays are equivalent.
Examples
--------
>>> arr1 = pd.array([1, 2, np.nan])
>>> arr2 = pd.array([1, 2, np.nan])
>>> arr1.equals(arr2)
True
F)
rn r rF rI rw r} r� r! r� r� )rf r� � equal_values�equal_nas rN �equalszExtensionArray.equals6 s� � �2 ��:��e�� $���^�U�+�� �:�:���� $��
��Y�#�e�*�
$�� �5�=�L��,��7� +� 2� 2�5� 9� ��y�y�{�U�Z�Z�\�1�H�� ��0�5�5�7�8� 8rQ c �@ � t t j | � |� S )a�
Pointwise comparison for set containment in the given values.
Roughly equivalent to `np.array([x in values for x in self])`
Parameters
----------
values : np.ndarray or ExtensionArray
Returns
-------
np.ndarray[bool]
Examples
--------
>>> arr = pd.array([1, 2, 3])
>>> arr.isin([1])
<BooleanArray>
[True, False, False]
Length: 3, dtype: boolean
)r' r� r� )rf r_ s rN r' zExtensionArray.isin_ s � �, �B�J�J�t�$�f�-�-rQ c �L � | j t � t j fS )a�
Return an array and missing value suitable for factorization.
Returns
-------
values : ndarray
An array suitable for factorization. This should maintain order
and be a supported dtype (Float64, Int64, UInt64, String, Object).
By default, the extension array is cast to object dtype.
na_value : object
The value in `values` to consider missing. This will be treated
as NA in the factorization routines, so it will be coded as
`-1` and not included in `uniques`. By default,
``np.nan`` is used.
Notes
-----
The values returned by this method are also used in
:func:`pandas.util.hash_pandas_object`. If needed, this can be
overridden in the ``self._hash_pandas_object()`` method.
Examples
--------
>>> pd.array([1, 2, 3])._values_for_factorize()
(array([1, 2, 3], dtype=object), nan)
)r� r� r� �nanrs s rN �_values_for_factorizez$ExtensionArray._values_for_factorizew s � �6 �{�{�6�"�B�F�F�*�*rQ c �v � | j � \ }}t |||�� \ }}| j || � }||fS )a�
Encode the extension array as an enumerated type.
Parameters
----------
use_na_sentinel : bool, default True
If True, the sentinel -1 will be used for NaN values. If False,
NaN values will be encoded as non-negative integers and will not drop the
NaN from the uniques of the values.
.. versionadded:: 1.5.0
Returns
-------
codes : ndarray
An integer NumPy array that's an indexer into the original
ExtensionArray.
uniques : ExtensionArray
An ExtensionArray containing the unique values of `self`.
.. note::
uniques will *not* contain an entry for the NA value of
the ExtensionArray if there are any missing values present
in `self`.
See Also
--------
factorize : Top-level factorize method that dispatches here.
Notes
-----
:meth:`pandas.factorize` offers a `sort` keyword as well.
Examples
--------
>>> idx1 = pd.PeriodIndex(["2014-01", "2014-01", "2014-02", "2014-02",
... "2014-03", "2014-03"], freq="M")
>>> arr, idx = idx1.factorize()
>>> arr
array([0, 0, 1, 1, 2, 2])
>>> idx
PeriodIndex(['2014-01', '2014-02', '2014-03'], dtype='period[M]')
)�use_na_sentinelr| )r� r& ra )rf r� r� r| �codesr� �
uniques_eas rN � factorizezExtensionArray.factorize� sK � �p �2�2�4�
��X�(� � �8�
���w� �*�*�7�D�9�
��j� � rQ a;
Repeat elements of a %(klass)s.
Returns a new %(klass)s where each element of the current %(klass)s
is repeated consecutively a given number of times.
Parameters
----------
repeats : int or array of ints
The number of repetitions for each element. This should be a
non-negative integer. Repeating 0 times will return an empty
%(klass)s.
axis : None
Must be ``None``. Has no effect but is accepted for compatibility
with numpy.
Returns
-------
%(klass)s
Newly created %(klass)s with repeated elements.
See Also
--------
Series.repeat : Equivalent function for Series.
Index.repeat : Equivalent function for Index.
numpy.repeat : Similar method for :class:`numpy.ndarray`.
ExtensionArray.take : Take arbitrary positions.
Examples
--------
>>> cat = pd.Categorical(['a', 'b', 'c'])
>>> cat
['a', 'b', 'c']
Categories (3, object): ['a', 'b', 'c']
>>> cat.repeat(2)
['a', 'a', 'b', 'b', 'c', 'c']
Categories (3, object): ['a', 'b', 'c']
>>> cat.repeat([1, 2, 3])
['a', 'b', 'b', 'c', 'c', 'c']
Categories (3, object): ['a', 'b', 'c']
�repeat)�klassc � � t j dd|i� t j t | � � j |� }| j
|� S )Nrd r� )r� �validate_repeatr� �arangerw r r� )rf �repeatsr� �inds rN r zExtensionArray.repeat sC � � ���2���~�.��i�i��D� �"�)�)�'�2���y�y��~� rQ )r� r� c � � t | � �)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 : Take elements from an array along an axis.
api.extensions.take : Take elements from an array.
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`.
Examples
--------
Here's an example implementation, which relies on casting the
extension array to object dtype. This uses the helper method
:func:`pandas.api.extensions.take`.
.. code-block:: python
def take(self, indices, allow_fill=False, fill_value=None):
from pandas.core.algorithms import take
# If the ExtensionArray is backed by an ndarray, then
# just pass that here instead of coercing to object.
data = self.astype(object)
if allow_fill and fill_value is None:
fill_value = self.dtype.na_value
# fill value should always be translated from the scalar
# type for the array, to the physical storage type for
# the data, before passing to take.
result = take(data, indices, fill_value=fill_value,
allow_fill=allow_fill)
return self._from_sequence(result, dtype=self.dtype)
r )rf �indicesr� r� s rN r� zExtensionArray.take s � �z "�$�'�'rQ c � � t | � �)a=
Return a copy of the array.
Returns
-------
ExtensionArray
Examples
--------
>>> arr = pd.array([1, 2, 3])
>>> arr2 = arr.copy()
>>> arr[0] = 2
>>> arr2
<IntegerArray>
[1, 2, 3]
Length: 3, dtype: Int64
r rs s rN rJ zExtensionArray.copyk s � �$ "�$�'�'rQ c �& � |� t |� �| dd S )a�
Return a view on the array.
Parameters
----------
dtype : str, np.dtype, or ExtensionDtype, optional
Default None.
Returns
-------
ExtensionArray or np.ndarray
A view on the :class:`ExtensionArray`'s data.
Examples
--------
This gives view on the underlying data of an ``ExtensionArray`` and is not a
copy. Modifications on either the view or the original ``ExtensionArray``
will be reflectd on the underlying data:
>>> arr = pd.array([1, 2, 3])
>>> arr2 = arr.view()
>>> arr[0] = 2
>>> arr2
<IntegerArray>
[2, 2, 3]
Length: 3, dtype: Int64
N)rm )rf rI s rN �viewzExtensionArray.view s � �@ � �%�e�,� ,��A�w�rQ c � � | j dkD r| j � S ddlm} || | j � d�� j d� }dt
| � j � d�}| j � }|� |� d |� �S )
Nr� r ��format_object_summaryF��indent_for_name�,
�<z>
�
) r� �_repr_2d�pandas.io.formats.printingr �
_formatter�rstriprn r� �_get_repr_footer)rf r �data�
class_name�footers rN �__repr__zExtensionArray.__repr__� s� � � �9�9�q�=��=�=�?� "�D�
%� �$�/�/�#�U�
�
�&��.�
� ��d��,�,�-�S�1�
��&�&�(�� � �d�V�2�f�X�.�.rQ c � � | j dkD r d| j � d| j � �S dt | � � d| j � �S )Nr� zShape: z , dtype: zLength: )r� r� rI rw rs s rN r z ExtensionArray._get_repr_footer� sC � � �9�9�q�=� �T�Z�Z�L� �$�*�*� �>� >��#�d�)��I�d�j�j�\�:�:rQ c �
� ddl m} | D �cg c]* } ||| j � d�� j d� ��, }}dj |� }dt | � j � d�}| j � }|� d |� d
|� �S c c}w ) Nr r Fr r z,
r �>z
[
z
]
)r r r r �joinrn r� r )rf r �x�linesr r r s rN r zExtensionArray._repr_2d� s� � �D� �
� ��
"�!�T�_�_�%6�� N� U� U��
� � �
� �z�z�%� ����d��,�,�-�Q�/�
��&�&�(�� � �U�4�&��f�X�6�6��
s �/B c � � |rt S t S )a�
Formatting function for scalar values.
This is used in the default '__repr__'. The returned formatting
function receives instances of your scalar type.
Parameters
----------
boxed : bool, default False
An indicated for whether or not your array is being printed
within a Series, DataFrame, or Index (True), or just by
itself (False). This may be useful if you want scalar values
to appear differently within a Series versus on its own (e.g.
quoted or not).
Returns
-------
Callable[[Any], str]
A callable that gets instances of the scalar type and
returns a string. By default, :func:`repr` is used
when ``boxed=False`` and :func:`str` is used when
``boxed=True``.
Examples
--------
>>> class MyExtensionArray(pd.arrays.NumpyExtensionArray):
... def _formatter(self, boxed=False):
... return lambda x: '*' + str(x) + '*' if boxed else repr(x) + '*'
>>> MyExtensionArray(np.array([1, 2, 3, 4]))
<MyExtensionArray>
[1*, 2*, 3*, 4*]
Length: 4, dtype: int64
)�str�repr)rf �boxeds rN r zExtensionArray._formatter� s � �D ��J�� rQ c � � | dd S )a�
Return a transposed view on this array.
Because ExtensionArrays are always 1D, this is a no-op. It is included
for compatibility with np.ndarray.
Returns
-------
ExtensionArray
Examples
--------
>>> pd.array([1, 2, 3]).transpose()
<IntegerArray>
[1, 2, 3]
Length: 3, dtype: Int64
Nrd )rf �axess rN � transposezExtensionArray.transpose� s
� �$ �A�w�rQ c �" � | j � S rc )r+ rs s rN �TzExtensionArray.T s � ��~�~� � rQ c � � | S )a
Return a flattened view on this array.
Parameters
----------
order : {None, 'C', 'F', 'A', 'K'}, default 'C'
Returns
-------
ExtensionArray
Notes
-----
- Because ExtensionArrays are 1D-only, this is a no-op.
- The "order" argument is ignored, is for compatibility with NumPy.
Examples
--------
>>> pd.array([1, 2, 3]).ravel()
<IntegerArray>
[1, 2, 3]
Length: 3, dtype: Int64
rd )rf �orders rN �ravelzExtensionArray.ravel s � �0 � rQ c � � t | � �)a�
Concatenate multiple array of this dtype.
Parameters
----------
to_concat : sequence of this type
Returns
-------
ExtensionArray
Examples
--------
>>> arr1 = pd.array([1, 2, 3])
>>> arr2 = pd.array([4, 5, 6])
>>> pd.arrays.IntegerArray._concat_same_type([arr1, arr2])
<IntegerArray>
[1, 2, 3, 4, 5, 6]
Length: 6, dtype: Int64
r )rL � to_concats rN r� z ExtensionArray._concat_same_type* rP rQ c �. � | j j S rc )rI r{ rs s rN r{ zExtensionArray._can_hold_naL s � ��z�z�&�&�&rQ �r� c �8 � t d|� d| j � �� �)as
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
Examples
--------
>>> arr = pd.array([1, 2, 3])
>>> arr._accumulate(name='cumsum')
<IntegerArray>
[1, 3, 6]
Length: 3, dtype: Int64
zcannot perform z with type )rm rI )rf �namer� r� s rN � _accumulatezExtensionArray._accumulateP s! � �L "�O�D�6��T�Z�Z�L�"Q�R�RrQ )r� �keepdimsc �� � t | |d� }|�2t dt | � j � d| j � d|� d�� � |dd|i|��}|rt j |g� }|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.
keepdims : bool, default False
If False, a scalar is returned.
If True, the result has dimension with size one along the reduced axis.
.. versionadded:: 2.1
This parameter is not required in the _reduce signature to keep backward
compatibility, but will become required in the future. If the parameter
is not found in the method signature, a FutureWarning will be emitted.
**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
Examples
--------
>>> pd.array([1, 2, 3])._reduce("min")
1
N�'z
' with dtype z does not support reduction 'r� rd )�getattrrV rn r� rI r� r� )rf r6 r� r8 r� r� r� s rN �_reducezExtensionArray._reducex s � �P �t�T�4�(�� �<���D��J�'�'�(�
�d�j�j�\� B/�/3�f�A�7��
� �.�V�.�v�.�� ��X�X�v�h�'�F��
rQ zClassVar[None]�__hash__c �, � t j | � S )aY
Specify how to render our entries in to_json.
Notes
-----
The dtype on the returned ndarray is not restricted, but for non-native
types that are not specifically handled in objToJSON.c, to_json is
liable to raise. In these cases, it may be safer to return an ndarray
of strings.
)r� r� rs s rN �_values_for_jsonz ExtensionArray._values_for_json� s � � �z�z�$� � rQ c �L � ddl m} | j � \ }} |||||�� S )a�
Hook for hash_pandas_object.
Default is to use the values returned by _values_for_factorize.
Parameters
----------
encoding : str
Encoding for data & key when strings.
hash_key : str
Hash_key for string key to encode.
categorize : bool
Whether to first categorize object arrays before hashing. This is more
efficient when the array contains duplicate values.
Returns
-------
np.ndarray[uint64]
Examples
--------
>>> pd.array([1, 2])._hash_pandas_object(encoding='utf-8',
... hash_key="1000000000000000",
... categorize=False
... )
array([ 6238072747940578789, 15839785061582574730], dtype=uint64)
r )�
hash_array)�encoding�hash_key�
categorize)�pandas.core.util.hashingrA r� )rf rB rC rD rA r_ �_s rN �_hash_pandas_objectz"ExtensionArray._hash_pandas_object� s/ � �< 8��.�.�0� ���� �X��Z�
�
rQ c � � | j � }t j t | � ft j �� }||fS )a�
Transform each element of list-like to a row.
For arrays that do not contain list-like elements the default
implementation of this method just returns a copy and an array
of ones (unchanged index).
Returns
-------
ExtensionArray
Array with the exploded values.
np.ndarray[uint64]
The original lengths of each list-like for determining the
resulting index.
See Also
--------
Series.explode : The method on the ``Series`` object that this
extension array method is meant to support.
Examples
--------
>>> import pyarrow as pa
>>> a = pd.array([[1, 2, 3], [4], [5, 6]],
... dtype=pd.ArrowDtype(pa.list_(pa.int64())))
>>> a._explode()
(<ArrowExtensionArray>
[1, 2, 3, 4, 5, 6]
Length: 6, dtype: int64[pyarrow], array([3, 1, 2], dtype=int32))
)r� rI )rJ r� �onesrw �uint64)rf r_ �countss rN �_explodezExtensionArray._explode� s3 � �> ���������D� �|�2�9�9�=���v�~� rQ c �| � | j dkD r | D �cg c] }|j � �� c}S t | � S c c}w )as
Return a list of the values.
These are each a scalar type, which is a Python scalar
(for str, int, float) or a pandas scalar
(for Timestamp/Timedelta/Interval/Period)
Returns
-------
list
Examples
--------
>>> arr = pd.array([1, 2, 3])
>>> arr.tolist()
[1, 2, 3]
r� )r� �tolist�list)rf r# s rN rN zExtensionArray.tolist
s9 � �$ �9�9�q�=�(,�-��1�A�H�H�J��-� -��D�z��� .s �9c � � t j t j t | � � |� }| j |� S rc )r� �deleter rw r� )rf �locr� s rN rQ zExtensionArray.delete s. � ��)�)�B�I�I�c�$�i� 0�#�6���y�y��!�!rQ c � � t |t | � � }t | � j |g| j �� }t | � j | d| || |d g� S )a�
Insert an item at the given position.
Parameters
----------
loc : int
item : scalar-like
Returns
-------
same type as self
Notes
-----
This method should be both type and dtype-preserving. If the item
cannot be held in an array of this type/dtype, either ValueError or
TypeError should be raised.
The default implementation relies on _from_sequence to raise on invalid
items.
Examples
--------
>>> arr = pd.array([1, 2, 3])
>>> arr.insert(2, -1)
<IntegerArray>
[1, 2, -1, 3]
Length: 4, dtype: Int64
r� N)r rw rn rO rI r� )rf rR rg �item_arrs rN �insertzExtensionArray.insert$ s[ � �< "�#�s�4�y�1����:�,�,�d�V�4�:�:�,�F���D�z�+�+�T�$�3�Z��4���:�,N�O�OrQ c �4 � t |� r|| }n|}|| |< y)a�
Analogue to np.putmask(self, mask, value)
Parameters
----------
mask : np.ndarray[bool]
value : scalar or listlike
If listlike, must be arraylike with same length as self.
Returns
-------
None
Notes
-----
Unlike np.putmask, we do not repeat listlike values with mismatched length.
'value' should either be a scalar or an arraylike with the same length
as self.
N)r )rf r� rp �vals rN �_putmaskzExtensionArray._putmaskH s# � �( �� ���+�C��C���T�
rQ c �Z � | j � }t |� r|| }n|}||| < |S )z�
Analogue to np.where(mask, self, value)
Parameters
----------
mask : np.ndarray[bool]
value : scalar or listlike
Returns
-------
same type as self
)rJ r )rf r� rp r� rW s rN �_wherezExtensionArray._wherec s7 � � ����� �� ����,�C��C����u�
��
rQ c �� � t j |� }| j t � } ||||j � �� | j || j �� }|| | |< y)z�
Replace values in locations specified by 'mask' using pad or backfill.
See also
--------
ExtensionArray.fillna
)r� r� r� N)r# �
get_fill_funcr� r� rJ rO rI )rf r� r� r� �func�npvaluesr� s rN �_fill_mask_inplacez!ExtensionArray._fill_mask_inplace{ s] � � �$�$�V�,���;�;�v�&��
�X�U�����5��(�(�����(�D�
� ��%��T�
rQ r �average�r� r� � na_optionr� �pctc �V � |dk7 rt �t | j � |||||�� S )z*
See Series.rank.__doc__.
r ra )rm r* r� )rf r� r� rb r� rc s rN �_rankzExtensionArray._rank� s9 � � �1�9�%� %�� � $� $� &��� � ��
�
rQ c � � | j g |�� }t j t j d� |� }|j |d�� }t || � r||j k7 rt d|� d�� �|S )z�
Create an ExtensionArray with the given shape and dtype.
See also
--------
ExtensionDtype.empty
ExtensionDtype.empty is the 'official' public version of this API.
r� r� Tr� z5Default 'empty' implementation is invalid for dtype='r: )rO r� � broadcast_to�intpr� r} rI rm )rL r� rI �obj�takerr� s rN �_emptyzExtensionArray._empty� sx � � � � ��5� �1��������� �U�3�����%�D��1���&�#�&�%�6�<�<�*?�%�G��w�a�P��
� �
rQ c �� � t j | j � � }t j | � }t j }t |||||� }t | � j
|� S )z�
Compute the quantiles of self for each quantile in `qs`.
Parameters
----------
qs : np.ndarray[float64]
interpolation: str
Returns
-------
same type as self
)r� r� r! r� r, rn rO )rf �qs�
interpolationr� r� r� �
res_valuess rN � _quantilezExtensionArray._quantile� sV � � �z�z�$�)�)�+�&���j�j�� ���V�V�
�'��T�:�r�=�Q�
��D�z�(�(��4�4rQ c � � 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.
)r� )r) )rf r� s rN �_modezExtensionArray._mode� s � �$ �D��(�(rQ c �D � t d� |D � � rt S t j | ||g|��i |��}|t ur|S d|v rt j | ||g|��i |��S |dk( r%t j
| ||g|��i |��}|t ur|S t j | ||g|��i |��S )Nc 3 �R K � | ] }t |t t t f� �� �! y �wrc )r} r r r )�.0r� s rN � <genexpr>z1ExtensionArray.__array_ufunc__.<locals>.<genexpr>� s# � �� �
�PV�u�J�u�y�(�L� A� B�PV�s �%'�out�reduce)r �NotImplementedr" �!maybe_dispatch_ufunc_to_dunder_op�dispatch_ufunc_with_out�dispatch_reduction_ufunc�default_array_ufunc)rf �ufuncr� �inputsr� r� s rN �__array_ufunc__z ExtensionArray.__array_ufunc__� s� � � �
�PV�
�
� "� !��<�<� �%��
�"(�
�,2�
�� � � '��M� �F�?� �4�4��e�V��&,��06��
� �X� � �7�7��e�V��&,��06��F� �^�+� �
��,�,�T�5�&�T�6�T�V�T�TrQ c � � t | ||�� S )a�
Map values using an input mapping or function.
Parameters
----------
mapper : function, dict, or Series
Mapping correspondence.
na_action : {None, 'ignore'}, default None
If 'ignore', propagate NA values, without passing them to the
mapping correspondence. If 'ignore' is not supported, a
``NotImplementedError`` should be raised.
Returns
-------
Union[ndarray, Index, ExtensionArray]
The output of the mapping function applied to the array.
If the function returns a tuple with more than one element
a MultiIndex will be returned.
)� na_action)r( )rf �mapperr� s rN �mapzExtensionArray.map� s � �( ��v��;�;rQ c � � ddl m} ddlm} |j |� } ||| |�� }
t | j |� rs|
j dvr?|
j |
j |
j t j t � d� | j t t j �� } nt d| j � �� � |
j | f|||d d
�|��} |
j |
j v r| S t | j |� r/| j }
|
j# � }|j% | |
� � S t �) a�
Dispatch GroupBy reduction or transformation operation.
This is an *experimental* API to allow ExtensionArray authors to implement
reductions and transformations. The API is subject to change.
Parameters
----------
how : {'any', 'all', 'sum', 'prod', 'min', 'max', 'mean', 'median',
'median', 'var', 'std', 'sem', 'nth', 'last', 'ohlc',
'cumprod', 'cumsum', 'cummin', 'cummax', 'rank'}
has_dropped_na : bool
min_count : int
ngroups : int
ids : np.ndarray[np.intp]
ids[i] gives the integer label for the group that self[i] belongs to.
**kwargs : operation-specific
'any', 'all' -> ['skipna']
'var', 'std', 'sem' -> ['ddof']
'cumprod', 'cumsum', 'cummin', 'cummax' -> ['skipna']
'rank' -> ['ties_method', 'ascending', 'na_option', 'pct']
Returns
-------
np.ndarray or ExtensionArray
r )� StringDtype)�WrappedCythonOp)�howr� �has_dropped_na)r r� F)r| z,function is not implemented for this dtype: N)� min_count�ngroups�comp_idsr� r� )�pandas.core.arrays.string_r� �pandas.core.groupby.opsr� �get_kind_from_howr} rI r� �_get_cython_functionr� r� r� r� r� rm �_cython_op_ndim_compat�cast_blocklistr� rO )rf r� r� r� r� �idsr� r� r� r� �opr^ ro rI �string_array_clss rN � _groupby_opzExtensionArray._groupby_op s- � �H ;�;� �0�0��5��
��4��
O�� �d�j�j�+� .��v�v�^�+��'�'����������&�9I�5�Q��}�}�V�b�f�f�}�=�H�%�>�t�z�z�l�K��
� /�R�.�.� �
� ����
� �
�
� �6�6�R�&�&� &� � � �d�j�j�+� .��J�J�E� $� 9� 9� ;� �#�2�2�:�U�2�K� K� &� %rQ )rI � Dtype | NonerJ r� )rI r6 �returnr= )rg r< r� r )rg r> r� r= )rg r; r� z
Self | Any�r� �None)r� �int)r� z
Iterator[Any])rg r� r� zbool | np.bool_)r� r� r� r2 )rI znpt.DTypeLike | NonerJ r� r| r� r� �
np.ndarray)r� r )r� r? ).)rI z
npt.DTypeLikerJ r� r� r� )rI r rJ r� r� rF )rI r3 rJ r� r� r2 )T)r� z)np.ndarray | ExtensionArraySupportsAnyAll)r� r� )r� r� )r� r� r� r@ r� r& r� r� )r� r� r� r� )
r� r8 r� r� r� rC rJ r� r� r= )
r� r7 r� �
int | Noner� z#Literal['inside', 'outside'] | NonerJ r� r� r= )NNNT)
rp zobject | ArrayLike | Noner� zFillnaOptions | Noner� r� rJ r� r� r= )r� r= )�first)r� z Literal['first', 'last', False]r� �npt.NDArray[np.bool_])r� N)r� r� r� r� r� rF )�leftN)rp z$NumpyValueArrayLike | ExtensionArrayr� zLiteral['left', 'right']r� zNumpySorter | Noner� z npt.NDArray[np.intp] | np.intp)r� r� r� r� )r_ r2 r� r� )r� ztuple[np.ndarray, Any])r� r� r� z!tuple[np.ndarray, ExtensionArray]rc )r zint | Sequence[int]r� zAxisInt | Noner� r= )r
rA r� r� r� r r� r= )rI r� r� r2 )r� r& )F)r( r� r� zCallable[[Any], str | None])r* r� r� rF )r� rF )�C)r/ z"Literal['C', 'F', 'A', 'K'] | Noner� rF )r2 zSequence[Self]r� r= )r6 r& r� r� r� rF )r6 r& r� r� r8 r� )rB r& rC r& rD r� r� znpt.NDArray[np.uint64])r� z#tuple[Self, npt.NDArray[np.uint64]])r� rO )rR r; r� r= )rR r� r� r= )r� r� r� r� )r� r� r� r= )r� r& r� r� r� r� r� r� )
r� r4 r� r& rb r& r� r� rc r� )r� r? rI r )rm znpt.NDArray[np.float64]rn r& r� r= )r� r� r� r= )r~ znp.ufuncr� r& ) r� r& r� r� r� r� r� r� r� znpt.NDArray[np.intp]r� r2 )Sr� �
__module__� __qualname__�__doc__�_typ�__pandas_priority__� classmethodrO rZ r] ra r
rh rq rt ry r� r� r� r r� r� �propertyrI r� r� r� r� r� r! r~ r� r� r� r� r� r� r� r� r% r� r+ r� r� r' r� r rD r r r r� rJ r
r r r r r+ r- r0 r� r r{ r7 r< �__annotations__r? rG rL rN rQ rU rX rZ r_ re rk rp rr r� r� r� rd rQ rN rF rF n s� � �N�d �D� �� �>B�QV� '� �'�8 � � � �> �/3�%� '� ,� '�;?� '� � '�< �'� �'�> � � � � � � � � (�B+S�Z(��(�(
(� � '+�� �>�>� "�#�"� �"� � "�
� "�P � (� � (� �
� �
� �#� �#� �
� �
� � (� � (�" � � � � � � � � � � � �E:�N(�8 �'� �'�! �L �$�!� 6
� �6
� � 6
�
� 6
�
�6
�p *�D *�D"
� #�"
� � "
�
� "
� �"
�
�"
�P !�:>��
]� �]� � ]�
8� ]� �
]�
�]�B ,0�'+� �� `�(�`� %�`� � `�
� `�
�
`�D"�$ 7>�=�3�=� �=�2:.�x>�, *0�%)� :A�3�:A� '�:A� #� :A�
(� :A�x'9�R.�0+�> !%�?!� �?!�
+�?!�F( � !���X �(�)�
�*�8�4�5� � 6� *� � !� � ](� �](� � ](�
� ](�
�
](�~(�("�P/� ;� 7�"$�T�( � � � ��4 �'� �'�B �'� �'� ,0�&S��&S�$(�&S� �&S�R ,0�%�2��2�$(�2�;?�2�n � � �#
� �#
�*-�#
�;?�#
� �#
�J! �F�,"�"P�H�6�0&��&�",�&�4I�&�
�&�* � � � ��
� �
� �
�
�
� �
� �
�0 �� ��,5�()�(U�4<�2J&� �J&� � J&�
� J&� �
J&� "�J&�
�J&rQ rF c �( � e Zd Zdd�dd�Zdd�dd�Zy)� ExtensionArraySupportsAnyAllTr4 c � � t | � �rc r r� s rN r z ExtensionArraySupportsAnyAll.anye �
� �!�$�'�'rQ c � � t | � �rc r r� s rN r� z ExtensionArraySupportsAnyAll.allh r� rQ N)r� r� r� r� )r� r� r� r r� rd rQ rN r� r� d s � �$(� (� %)� (rQ r� c �v � e Zd ZdZed� � Zed d�� Zed� � Zed d�� Zed� � Z ed d�� Z
y)
�ExtensionOpsMixinz�
A base class for linking the operators to their dunder names.
.. note::
You may want to set ``__array_priority__`` if you want your
implementation to be called when involved in binary operations
with NumPy arrays.
c � � t | � �rc r �rL r� s rN �_create_arithmetic_methodz+ExtensionOpsMixin._create_arithmetic_methodw �
� �!�#�&�&rQ c �0 � t | d| j t j � � t | d| j t j
� � t | d| j t j � � t | d| j t j � � t | d| j t j � � t | d| j t j � � t | d| j t j � � t | d| j t j � � t | d | j t j � � t | d
| j t j � � t | d | j t j � � t | d | j t j � � t | d
| j t j � � t | d| j t j" � � t | d| j t$ � � t | d| j t j& � � y )N�__add__�__radd__�__sub__�__rsub__�__mul__�__rmul__�__pow__�__rpow__�__mod__�__rmod__� __floordiv__�
__rfloordiv__� __truediv__� __rtruediv__�
__divmod__� __rdivmod__)�setattrr� �operator�addr$ �radd�sub�rsub�mul�rmul�pow�rpow�mod�rmod�floordiv� rfloordiv�truediv�rtruediv�divmod�rdivmod�rL s rN �_add_arithmetic_opsz%ExtensionOpsMixin._add_arithmetic_ops{ s� � ���Y�� =� =�h�l�l� K�L���Z��!>�!>�y�~�~�!N�O���Y�� =� =�h�l�l� K�L���Z��!>�!>�y�~�~�!N�O���Y�� =� =�h�l�l� K�L���Z��!>�!>�y�~�~�!N�O���Y�� =� =�h�l�l� K�L���Z��!>�!>�y�~�~�!N�O���Y�� =� =�h�l�l� K�L���Z��!>�!>�y�~�~�!N�O���^�S�%B�%B�8�CT�CT�%U�V�� � �#�"?�"?� �@S�@S�"T�
� ��]�C�$A�$A�(�BR�BR�$S�T���^�S�%B�%B�9�CU�CU�%V�W���\�3�#@�#@��#H�I���]�C�$A�$A�)�BS�BS�$T�UrQ c � � t | � �rc r r� s rN �_create_comparison_methodz+ExtensionOpsMixin._create_comparison_method� r� rQ c � � t | d| j t j � � t | d| j t j � � t | d| j t j
� � t | d| j t j � � t | d| j t j � � t | d| j t j � � y )Nr� r� �__lt__�__gt__�__le__�__ge__) r� r� r� �eq�ne�lt�gt�le�ger� s rN �_add_comparison_opsz%ExtensionOpsMixin._add_comparison_ops� s� � ���X�s� <� <�X�[�[� I�J���X�s� <� <�X�[�[� I�J���X�s� <� <�X�[�[� I�J���X�s� <� <�X�[�[� I�J���X�s� <� <�X�[�[� I�J���X�s� <� <�X�[�[� I�JrQ c � � t | � �rc r r� s rN �_create_logical_methodz(ExtensionOpsMixin._create_logical_method� r� rQ c � � t | d| j t j � � t | d| j t j
� � t | d| j t j � � t | d| j t j � � t | d| j t j � � t | d| j t j � � y )N�__and__�__rand__�__or__�__ror__�__xor__�__rxor__)
r� r� r� �and_r$ �rand_�or_�ror_�xor�rxorr� s rN �_add_logical_opsz"ExtensionOpsMixin._add_logical_ops� s� � ���Y�� :� :�8�=�=� I�J���Z��!;�!;�I�O�O�!L�M���X�s� 9� 9�(�,�,� G�H���Y�� :� :�9�>�>� J�K���Y�� :� :�8�<�<� H�I���Z��!;�!;�I�N�N�!K�LrQ Nr� ) r� r� r� r� r� r� r� r� r� r� r� rd rQ rN r� r� l s� � �� �'� �'� �V� �V�( �'� �'� �K� �K� �'� �'� �M� �MrQ r� c �D � e Zd ZdZeddd�� Zed� � Zed� � Zy)�ExtensionScalarOpsMixina�
A mixin for defining ops on an ExtensionArray.
It is assumed that the underlying scalar objects have the operators
already defined.
Notes
-----
If you have defined a subclass MyExtensionArray(ExtensionArray), then
use MyExtensionArray(ExtensionArray, ExtensionScalarOpsMixin) to
get the arithmetic operators. After the definition of MyExtensionArray,
insert the lines
MyExtensionArray._add_arithmetic_ops()
MyExtensionArray._add_comparison_ops()
to link the operators to your class.
.. note::
You may want to set ``__array_priority__`` if you want your
implementation to be called when involved in binary operations
with NumPy arrays.
Nc �P ���� ���fd�}d�j � d�}t ||| � S )a
A class method that returns a method that will correspond to an
operator for an ExtensionArray subclass, by dispatching to the
relevant operator defined on the individual elements of the
ExtensionArray.
Parameters
----------
op : function
An operator that takes arguments op(a, b)
coerce_to_dtype : bool, default True
boolean indicating whether to attempt to convert
the result to the underlying ExtensionArray dtype.
If it's not possible to create a new ExtensionArray with the
values, an ndarray is returned instead.
Returns
-------
Callable[[Any, Any], Union[ndarray, ExtensionArray]]
A method that can be bound to a class. When used, the method
receives the two arguments, one of which is the instance of
this class, and should return an ExtensionArray or an ndarray.
Returning an ndarray may be necessary when the result of the
`op` cannot be stored in the ExtensionArray. The dtype of the
ndarray uses NumPy's normal inference rules.
Examples
--------
Given an ExtensionArray subclass called MyExtensionArray, use
__add__ = cls._create_method(operator.add)
in the class definition of MyExtensionArray to create the operator
for addition, that will be based on the operator implementation
of the underlying elements of the ExtensionArray
c �, �� � � fd�}t |t t t f� rt S � } ||� }t ||� D ��cg c] \ }} �
||� �� }}}� � � fd�}�
j dv rt |� \ }} ||� ||� fS ||� S c c}}w )Nc �d �� t | t � s t | � r| }|S | gt �� z }|S rc )r} rF r rw )�param�ovaluesrf s �rN �convert_valueszNExtensionScalarOpsMixin._create_method.<locals>._binop.<locals>.convert_values� s7 �� � �e�^�4� �U�8K� #�G� �� %�g��D� � 1�G� �rQ c �� �� �rDt | �j d�� }t |t �� � st j
| � }|S t j
| ��� }|S )NF)�
same_dtyper� )r rI r} rn r� r� )r� �res�coerce_to_dtype� result_dtyperf s ���rN �_maybe_convertzNExtensionScalarOpsMixin._create_method.<locals>._binop.<locals>._maybe_convert
sT �� �"� 6�c�4�:�:�RW�X�C�%�c�4��:�6� �j�j��o�� �
� �*�*�S� �=�C��
rQ > r� r� )r} r r r ry �zipr� ) rf r� r� �lvalues�rvaluesr� r� r r r r� r s ` ���rN �_binopz6ExtensionScalarOpsMixin._create_method.<locals>._binop� s� �� �
� �%�)�X�|�!D�E�%�%��G�$�U�+�G� +.�g�w�*?�@�*?���A�2�a��8�*?�C�@�
� �{�{� 3�3��C�y���1�%�a�(�.��*;�;�;�!�#�&� &��' As �B�__)r� r
)rL r� r r r �op_names ``` rN �_create_methodz&ExtensionScalarOpsMixin._create_method� s- �� �P$ '�L �r�{�{�m�2�&�� ���#�6�6rQ c �$ � | j |� S rc )r r� s rN r� z1ExtensionScalarOpsMixin._create_arithmetic_method
s � ��!�!�"�%�%rQ c �2 � | j |dt �� S )NF)r r )r r� r� s rN r� z1ExtensionScalarOpsMixin._create_comparison_method
s � ��!�!�"�e�$�!�O�OrQ )TN)r r� )r� r� r� r� r� r r� r� rd rQ rN r� r� � sH � ��2 �N7� �N7�` �&� �&� �P� �PrQ r� )br� �
__future__r r� �typingr r r r r r r
rX �numpyr� � pandas._libsr r� r �
pandas.compatr
�pandas.compat.numpyr r� �
pandas.errorsr �pandas.util._decoratorsr r r �pandas.util._exceptionsr �pandas.util._validatorsr r r �pandas.core.dtypes.castr �pandas.core.dtypes.commonr r r �pandas.core.dtypes.dtypesr �pandas.core.dtypes.genericr r r �pandas.core.dtypes.missingr! � pandas.corer" r# r$ �pandas.core.algorithmsr% r& r' r( r) r* r+ � pandas.core.array_algos.quantiler, �pandas.core.missingr- �pandas.core.sortingr. r/ �collections.abcr0 r1 �pandas._typingr2 r3 r4 r5 r6 r7 r8 r9 r: r; r< r= r> r? r@ rA rB �pandasrC rD r� rF r� r� r� rd rQ rN �<module>r& s ��� #� �� � � � �� ,� .� -�� �
5�� � @�� �
5�� �
,�� �
� � � @� 3��
��
� � � � �( �/1� �n� 1�s#&� s#&�lG(�>� (�<M� <M�~qP� /� qPrQ
Sindbad File Manager Version 1.0, Coded By Sindbad EG ~ The Terrorists