U
|
¬ý°dð/ ã @ sò d dl mZ d dlZd dlmZmZ d dlZd dlm Z m
|
Z d dl m Z mZmZ d dlmZmZ d dlmZ d dlmZ d d lmZ d d
|
lmZ d d lm Z m Z er¸d dl Z d d l m Z eG d d de Z ddddddZ!G dd de Z"dS )é )Ú annotationsN)Ú TYPE_CHECKINGÚcast)ÚlibÚmissing)ÚDtypeÚDtypeObjÚtype_t)Ú is_list_likeÚis_numeric_dtype)Úregister_extension_dtype)Úisna)Úops)Úmasked_accumulations)ÚBaseMaskedArrayÚBaseMaskedDtype)Únptc @ s e Zd ZdZdZeddddZedddd Zed
|
dd d Ze d dddZ
|
ddddZ eddddZ eddddZ dddddZd S ) Ú BooleanDtypeaI
|
Extension dtype for boolean data.
|
|
.. warning::
|
|
BooleanDtype is considered experimental. The implementation and
|
parts of the API may change without warning.
|
|
Attributes
|
----------
|
None
|
|
Methods
|
-------
|
None
|
|
Examples
|
--------
|
>>> pd.BooleanDtype()
|
BooleanDtype
|
ÚbooleanÚtype©Úreturnc C s t jS ©N)ÚnpÚbool_©Úself© r úQd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/core/arrays/boolean.pyr E s zBooleanDtype.typeÚstrc C s dS )NÚbr r r r r ÚkindI s zBooleanDtype.kindznp.dtypec C s
|
t d¡S )NÚbool)r Údtyper r r r Ú numpy_dtypeM s zBooleanDtype.numpy_dtypeztype_t[BooleanArray]c C s t S )zq
|
Return the array type associated with this dtype.
|
|
Returns
|
-------
|
type
|
)Ú BooleanArray)Úclsr r r Úconstruct_array_typeQ s z!BooleanDtype.construct_array_typec C s dS )Nr r r r r r Ú__repr__\ s zBooleanDtype.__repr__r" c C s dS ©NTr r r r r Ú _is_boolean_ s zBooleanDtype._is_booleanc C s dS r) r r r r r Ú _is_numericc s zBooleanDtype._is_numericz$pyarrow.Array | pyarrow.ChunkedArrayr% )Úarrayr c
|
C s ddl }|j| ¡ kr(td|j dt||jr<|g}n|j}g }|D ]}| ¡ }|jj |jt
|
|d|d g|j dj dd}|j dkrÀ|jj |jt
|
|d|d g|j dj dd}| }ntjt
|
|td }t||} | | ¡ qJ|sttjg tjd tjg tjd S t |¡S dS )
|
zI
|
Construct BooleanArray from pyarrow Array/ChunkedArray.
|
r Nz$Expected array of boolean type, got z insteadé )ÚoffsetF)Zzero_copy_only©r# )Úpyarrowr r Ú TypeErrorÚ
|
isinstanceÚArrayÚchunksÚbuffersr% Z from_buffersÚlenr. Zto_numpyZ
|
null_countr Úzerosr" Úappendr, Z_concat_same_type)
|
r r, r0 r4 ÚresultsZarrZbuflistÚdataÚmaskZbool_arrr r r Ú__from_arrow__g sH
|
ÿþ
|
|
ÿþ
|
ÿzBooleanDtype.__from_arrow__N)Ú__name__Ú
|
__module__Ú __qualname__Ú__doc__ÚnameÚpropertyr r! r$ Ú classmethodr' r( r* r+ r< r r r r r ) s
|
r Fr" ú tuple[np.ndarray, np.ndarray])Úcopyr c C s@ t | trD|dk rtd| j| j } }|r<| ¡ } | ¡ }| |fS d}t | tjrp| jtj krp|rl| ¡ } n t | tjræt
|
| jræt | }tj t | td}| | t¡|| < t || | j¡| | k¡sàtd|} n¦tj| td}tj|dd}d}|d| kr tdtd t |}tj t | td} || t¡| | < ||krt | | t¡|| t¡k¡std|dkr²|dkr²tj | jtd}np|dkrÂ|}n`t |tjr|jtj kr|dk rò||B }n|r"| ¡ }n tj|td}|dk r"||B }| j|jkr8td
|
| |fS ) a
|
Coerce the input values array to numpy arrays with a mask.
|
|
Parameters
|
----------
|
values : 1D list-like
|
mask : bool 1D array, optional
|
copy : bool, default False
|
if True, copy the input
|
|
Returns
|
-------
|
tuple of (values, mask)
|
Nz'cannot pass mask for BooleanArray inputr/ z Need to pass bool-like valuesT©Úskipna)ZfloatingÚintegerzmixed-integer-float)r Úemptyznpt.NDArray[np.bool_]z&values.shape and mask.shape must match)r2 r% Ú
|
ValueErrorÚ_dataÚ_maskrE r Úndarrayr# r r r r7 r6 r" ÚastypeÚallr1 ÚasarrayÚobjectr Z infer_dtyper ÚfloatÚshaper, )Úvaluesr; rE Z mask_valuesZ values_boolZ values_objectZinferred_dtypeZ integer_liker r r Úcoerce_to_array sh
|
ÿ
|
ÿÿÿ
|
|
|
|
|
rU c sÞ e Zd ZdZdZdZdZdddddhZd d
|
d d d hZd.ddddd fdd Z e
|
ddddZ e ddddddddd d d d d d Z ejejeejfZe dd d!dd"d#d$d%Zd&d' Zdd(d)dd*d+d,d-Z ZS )/r% a=
|
Array of boolean (True/False) data with missing values.
|
|
This is a pandas Extension array for boolean data, under the hood
|
represented by 2 numpy arrays: a boolean array with the data and
|
a boolean array with the mask (True indicating missing).
|
|
BooleanArray implements Kleene logic (sometimes called three-value
|
logic) for logical operations. See :ref:`boolean.kleene` for more.
|
|
To construct an BooleanArray from generic array-like input, use
|
:func:`pandas.array` specifying ``dtype="boolean"`` (see examples
|
below).
|
|
.. warning::
|
|
BooleanArray is considered experimental. The implementation and
|
parts of the API may change without warning.
|
|
Parameters
|
----------
|
values : numpy.ndarray
|
A 1-d boolean-dtype array with the data.
|
mask : numpy.ndarray
|
A 1-d boolean-dtype array indicating missing values (True
|
indicates missing).
|
copy : bool, default False
|
Whether to copy the `values` and `mask` arrays.
|
|
Attributes
|
----------
|
None
|
|
Methods
|
-------
|
None
|
|
Returns
|
-------
|
BooleanArray
|
|
Examples
|
--------
|
Create an BooleanArray with :func:`pandas.array`:
|
|
>>> pd.array([True, False, None], dtype="boolean")
|
<BooleanArray>
|
[True, False, <NA>]
|
Length: 3, dtype: boolean
|
FTÚTrueÚTRUEÚtrueÚ1z1.0ÚFalseÚFALSEÚfalseÚ0z0.0z
|
np.ndarrayr" ÚNone)rT r; rE r c s> t |tjr|jtjks tdt | _t j |||d d S )NzIvalues should be boolean numpy array. Use the 'pd.array' function instead©rE )
|
r2 r rM r# r r1 r Ú_dtypeÚsuperÚ__init__)r rT r; rE ©Ú __class__r r rb & s ÿzBooleanArray.__init__r r c C s | j S r )r` r r r r r# 1 s zBooleanArray.dtypeN)r# rE Ú true_valuesÚ false_valuesz list[str]z Dtype | Nonezlist[str] | None)Ústringsr# rE re rf r c st | j |p g ¡| j |p g ¡ dd fdd }tj|td}t|}tt||| || < | j |||dS )Nr" r c s* | kr dS | krdS t | dd S )NTFz cannot be cast to bool)rJ )Ús©Zfalse_values_unionZtrue_values_unionr r Ú
|
map_stringB s
|
z:BooleanArray._from_sequence_of_strings.<locals>.map_stringr/ )r# rE )
|
Ú _TRUE_VALUESÚunionÚ _FALSE_VALUESr r, rQ r ÚlistÚmapZ_from_sequence) r& rg r# rE re rf rj Zscalarsr; r ri r Ú_from_sequence_of_strings5 s
|
z&BooleanArray._from_sequence_of_stringsr_ r rD )r# rE r c C s |r|dkst t||dS )Nr r_ )ÚAssertionErrorrU )r& Úvaluer# rE r r r Ú_coerce_to_arrayQ s z BooleanArray._coerce_to_arrayc C s< |j dkstt |¡}d }t|tr6|j|j }}nNt|rpt j
|
|dd}|j dkr^t dt |dd\}}nt|t jr| ¡ }|r²|tjk r²t |¡s²tdt|j d |sÎt| t|krÎtd
|
|j d kròt | j|| j|¡\}}n>|j d krt | j|| j|¡\}}nt | j|| j|¡\}}| ||¡S ) N> ÚxorÚrxorÚrand_Úor_Úand_Úror_r" r/ r- z(can only perform ops with 1-d structuresFr_ z+'other' should be pandas.NA or a bool. Got z instead.zLengths must match> ry rw > rv rx ) r= rq r Z is_scalarr2 r% rK rL r
|
r rP ÚndimÚNotImplementedErrorrU r ÚitemÚ
|
libmissingZNAZis_boolr1 r r6 rJ r Z kleene_orZ
|
kleene_andZ
|
kleene_xorZ_maybe_mask_result)r ÚotherÚopZother_is_scalarr; Úresultr r r Ú_logical_methodY s0
|
|
|
ÿ
|
z BooleanArray._logical_methodrF r r )rA rG r c K s| | j }| j}|dkrJtt|}|||fd|i|\}}t| ||ddS ddlm} || t¡|j |fd|i|S d S )N)ZcumminZcummaxrG Fr_ r )Ú IntegerArray)
|
rK rL Úgetattrr r Zpandas.core.arraysr rN ÚintÚ _accumulate)r rA rG Úkwargsr: r; r r r r r r
| s
|
ÿÿÿzBooleanArray._accumulate)F)r= r> r? r@ Z_internal_fill_valueZ _truthy_valueZ _falsey_valuerk rm rb rB r# rC rp r rM ÚnumbersÚNumberr" r Z_HANDLED_TYPESrs r r
Ú __classcell__r r rc r r% è s. 4 ÿ ù ÿ$ÿr% )NF)#Ú
|
__future__r r Útypingr r Únumpyr Z pandas._libsr r r} Zpandas._typingr r r Zpandas.core.dtypes.commonr
|
r Zpandas.core.dtypes.dtypesr Zpandas.core.dtypes.missingr Z pandas.corer Zpandas.core.array_algosr Zpandas.core.arrays.maskedr r r0 r r rU r% r r r r Ú<module> s( h ÿW
|
