U
|
¬ý°d7 ã @ sâ d Z ddlmZ ddlmZmZ ddlZddlm Z ddl
|
m Z m Z ddl mZ ddlmZ dd lmZ dd
|
lmZ erdd lmZ d d dddddZdd dddddZddddddZd d d d d Zdd d d!d"ZdS )#z&
|
EA-compatible analogue to np.putmask
|
é )Ú annotations)Ú TYPE_CHECKINGÚAnyN)Úlib)Ú ArrayLikeÚnpt)Únp_version_under1p21)Úinfer_dtype_from)Ú is_list_like)ÚExtensionArray)Ú
|
MultiIndexr znpt.NDArray[np.bool_]r ÚNone)ÚvaluesÚmaskÚvalueÚreturnc C s~ t | tjr<| jtkr t |¡r<t |tjrlt |j| j¡slt|rbt |t | krb|| | |< qz|| |< nt
|
| ||¡ dS )aD
|
ExtensionArray-compatible implementation of np.putmask. The main
|
difference is we do not handle repeating or truncating like numpy.
|
|
Parameters
|
----------
|
values: np.ndarray or ExtensionArray
|
mask : np.ndarray[bool]
|
We assume extract_bool_array has already been called.
|
value : Any
|
N) Ú
|
isinstanceÚnpZndarrayÚdtypeÚobjectr Z is_scalarZcan_castr
|
ÚlenÚputmask)r r r © r úVd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/core/array_algos/putmask.pyÚputmask_inplace s
|
ÿþþ
|
úú
|
r z
|
np.ndarray)r r Únewr c C sÌ t rt| | ¡ |}t|dddkr4|j| jdd}| ¡ }|dkrºt|rºt|dddkrºt |¡}||d krt | ||¡ qÈ|jd |d ks |d dkr°t
|
| ||¡ qÈt dnt
|
| ||¡ dS ) zâ
|
np.putmask will truncate or repeat if `new` is a listlike with
|
len(new) != len(values). We require an exact match.
|
|
Parameters
|
----------
|
values : np.ndarray
|
mask : np.ndarray[bool]
|
new : Any
|
Úndimr é F)Úcopyéÿÿÿÿz-cannot assign mismatch length to masked arrayN) r Úsetitem_datetimelike_compatÚsumÚgetattrZastyper r
|
r ÚshapeZplacer Ú
|
ValueError)r r r Znlocsr# r r r Úputmask_without_repeat> s
|
|
r% zArrayLike | MultiIndexz"tuple[npt.NDArray[np.bool_], bool])r r r c C s. t |}|j| jkr td| ¡ }||fS )zG
|
Validate mask and check if this putmask operation is a no-op.
|
z,putmask: mask and data must be the same size)Úextract_bool_arrayr# r$ Úany)r r Znoopr r r Úvalidate_putmaskh s
|
|
r( )r r c C s* t | tr| jtdd} tj| td} | S )zP
|
If we have a SparseArray or BooleanArray, convert it to ndarray[bool].
|
F)r Zna_value)r )r r Zto_numpyÚboolr Zasarray)r r r r r& v s
|
r& Úint)r Únum_setc C sP | j tkrLt|dd\}}t|tj rL|jdkrLt|sD|g| }nt|}|S )z
|
Parameters
|
----------
|
values : np.ndarray
|
num_set : int
|
For putmask, this is mask.sum()
|
other : Any
|
T)Z pandas_dtype)ÚmÚM)r r r r r Úkindr
|
Úlist)r r+ Úotherr Ú_r r r r s
|
r ) Ú__doc__Ú
|
__future__r Útypingr r Únumpyr Z pandas._libsr Zpandas._typingr r Z pandas.compatr Zpandas.core.dtypes.castr Zpandas.core.dtypes.commonr
|
Zpandas.core.arraysr Zpandasr r r% r( r& r r r r r Ú<module> s !*
|
