U
|
¬ý°d¥ ã @ s¬ d Z ddlmZ ddlZddlZddlmZmZ ddlZ ddl
|
m Z m Z m Z ddlmZmZmZ ddlmZ dd dd
|
d d Zd dddd dddZd ddddddZdS )z4
|
Methods used by Block.replace and related methods.
|
é )Ú annotationsN)ÚAnyÚPattern)Ú ArrayLikeÚScalarÚnpt)Úis_reÚis_re_compilableÚ is_scalar)ÚisnaÚboolr )ÚregexÚ
|
to_replaceÚreturnc C s0 t |r d} | ot|} | o*t |¡jdk} | S )zG
|
Decide whether to treat `to_replace` as a regular expression.
|
TÚ )r r ÚreÚcompileÚpattern)r r © r úVd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/core/array_algos/replace.pyÚshould_use_regex s
|
r r úScalar | Patternznpt.NDArray[np.bool_])ÚaÚbr Úmaskr c s² t r| S dddddd}|r.t| s< fdd}nt fd d¡}t| tjrb| | } || }t|tjr¢|d
|
k r¢tj|jtjd }t |||¡ |}||| |S ) a
|
Compare two array-like inputs of the same shape or two scalar values
|
|
Calls operator.eq or re.search, depending on regex argument. If regex is
|
True, perform an element-wise regex matching.
|
|
Parameters
|
----------
|
a : array-like
|
b : scalar or regex pattern
|
regex : bool
|
mask : np.ndarray[bool]
|
|
Returns
|
-------
|
mask : array-like of bool
|
zArrayLike | boolr r )Úresultr r c S sb t | r^t|tjr^t|jt|jg}d|j d|d< tdt|d dt|d dS )z
|
Raises an error if the two arrays (a,b) cannot be compared.
|
Otherwise, returns the comparison result as expected.
|
zndarray(dtype=ú)r zCannot compare types z and é N) r
|
Ú
|
isinstanceÚnpÚndarrayÚtypeÚ__name__ÚdtypeÚ TypeErrorÚrepr)r r r Ú
|
type_namesr r r Ú_check_comparison_typesB s ÿz8compare_or_regex_search.<locals>._check_comparison_typesc s t | ¡S ©N)ÚoperatorÚeq©Úx©r r r Ú<lambda>T ó z)compare_or_regex_search.<locals>.<lambda>c s, t | tr(t ttfr(tt | ¡S dS )NF)r Ústrr r r Úsearchr+ r- r r r. W s
|
ÿ ÿN)r# )
|
r r r Ú vectorizer r ÚzerosÚshapeZbool_Zplace)r r r r r' Úopr Útmpr r- r Úcompare_or_regex_search+ s"
|
ÿ r7 z
|
re.Patternz npt.NDArray[np.bool_] | NoneÚNone)ÚvaluesÚrxr r c sp t stts" fdd}n fdd}tj|tjgd}|dkr\|| | dd
< n|| | | |< dS )zË
|
Parameters
|
----------
|
values : ArrayLike
|
Object dtype.
|
rx : re.Pattern
|
value : Any
|
mask : np.ndarray[bool], optional
|
|
Notes
|
-----
|
Alters values in-place.
|
c s0 t r(t| tr( | ¡d k r$S | S | S d S r( )r r r0 r1 ©Ús©r: Úvaluer r Ú re_replacer s z"replace_regex.<locals>.re_replacerc s& t r t| tr | ¡S | S d S r( )r r r0 Úsubr; r= r r r? s )ZotypesN)r r r0 r r2 Zobject_)r9 r: r> r r? Úfr r= r Ú replace_regexm s rB )Ú__doc__Ú
|
__future__r r) r Útypingr r Únumpyr Zpandas._typingr r r Zpandas.core.dtypes.commonr r r
|
Zpandas.core.dtypes.missingr r r7 rB r r r r Ú<module> s B
|
