U
|
¬ý°d
H ã 2 @ s² d Z ddlmZ ddlZddlmZmZ ddlZddlm Z ddl
|
Z
|
ddl m Z m Z mZmZmZmZmZmZmZmZ ddlZddlZddlmZ ddlm Z m Z m Z m Z m Z dd l!m"Z" dd
|
l#m$Z$m%Z%m&Z&m'Z' dd l(m)Z)m*Z*m+Z+ dd l,m-Z- dd l.m/Z/ e rddl0m1Z1 dd Z2dd Z3dddddZ4dd Z5dd Z6dd d d Z7dd d d Z8dd d!d"Z9dd d#d$Z:d%d d&d'Z;edd)d*d+d,d-d.Z<edd/d*d0d,d1d.Z<dd/d*d0d,d2d.Z<dd3d*d+d4d5d6Z=d7d8 Z>d9d:d;d<d=Z?dd d>d?Z@dd d@dAZAdBd dCdDZBd%ddEdFdGZCdHdI ZDdJdK ZEdLdM ZFedNdNdOdPdQZGedRdSdOdTdQZGddUdVdWdQZGdXdYdZd[d\ZHd]d^ ZId_d`dadbdcZJejKdddedfdgdhZLdidjdkdldmZMejNejNejOejPjQejRejSjQiZTejNdnejOdoejRdpejUdqejVdrejNdnejWdnejXdsejYdsejZdtej[dtej\duej]duej^dvej_dvej`dwejadwejOdoejbdoejRdpejcdpejddxejedxejfdyejgdyiZhdzd{d|d}d~Zidd Zjddddd
ZkdS )zh
|
Misc tools for implementing data structures
|
|
Note: pandas.core.common is *not* part of the public API.
|
é )Ú annotationsN)ÚabcÚ defaultdict)Úpartial)
|
Ú TYPE_CHECKINGÚAnyÚCallableÚ
|
CollectionÚ GeneratorÚHashableÚIterableÚSequenceÚcastÚoverload)Úlib)Ú AnyArrayLikeÚ ArrayLikeÚNpDtypeÚ RandomStateÚT)Ú'construct_1d_object_array_from_listlike)Ú is_array_likeÚ is_bool_dtypeÚis_extension_array_dtypeÚ
|
is_integer)ÚABCExtensionArrayÚABCIndexÚ ABCSeries)Úiterable_not_string)Úisna)ÚIndexc c s, | D ]"}t |r t|E dH q|V qdS )a
|
Flatten an arbitrarily nested sequence.
|
|
Parameters
|
----------
|
line : sequence
|
The non string sequence to flatten
|
|
Notes
|
-----
|
This doesn't consider strings sequences.
|
|
Returns
|
-------
|
flattened : generator
|
N)r Úflatten)ÚlineÚelement© r$ úId:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/core/common.pyr! < s r! c C sL | d j }| dd
D ]0}z|j |kr*d }W q tk
|
rD d }Y qX q|S )Nr é )ÚnameÚ
|
ValueError)Úobjsr' Úobjr$ r$ r% Úconsensus_name_attrT s
|
|
r+ r Úbool)ÚkeyÚreturnc C sº t | ttjtfs$t| rt| jr| jtjkrrt | ¡}t
|
|¡snd}t
|
|¡dkrjt | ¡ rjt|dS dS t| jr¶dS n4t | tr¶t| dkr¶t| tk r¬t| } t
|
| ¡S dS )aË
|
Check whether `key` is a valid boolean indexer.
|
|
Parameters
|
----------
|
key : Any
|
Only list-likes may be considered boolean indexers.
|
All other types are not considered a boolean indexer.
|
For array-like input, boolean ndarrays or ExtensionArrays
|
with ``_is_boolean`` set are considered boolean indexers.
|
|
Returns
|
-------
|
bool
|
Whether `key` is a valid boolean indexer.
|
|
Raises
|
------
|
ValueError
|
When the array is an object-dtype ndarray or ExtensionArray
|
and contains missing values.
|
|
See Also
|
--------
|
check_array_indexer : Check that `key` is a valid array to index,
|
and convert to an ndarray.
|
z=Cannot mask with non-boolean array containing NA / NaN valuesÚbooleanFTr )Ú
|
isinstancer ÚnpÚndarrayr r r ÚdtypeÚobject_Úasarrayr Z is_bool_arrayZ infer_dtyper Úanyr( r ÚlistÚlenÚtypeZ is_bool_list)r- Z key_arrayZna_msgr$ r$ r% Úis_bool_indexer_ s* ÿÿ
|
|
|
|
|
r: c C s t | ¡r| ¡ rtd| S )z®
|
Disallow indexing with a float key, even if that key is a round number.
|
|
Parameters
|
----------
|
val : scalar
|
|
Returns
|
-------
|
outval : scalar
|
zYIndexing with a float is no longer supported. Manually convert to an integer key instead.)r Zis_floatr Ú
|
IndexError)Úvalr$ r$ r% Úcast_scalar_indexer s
|
þr= c G s dd | D S )zL
|
Returns a generator consisting of the arguments that are not None.
|
c s s | ]}|d k r|V qd S ©Nr$ ©Ú.0Úargr$ r$ r% Ú <genexpr>° s znot_none.<locals>.<genexpr>r$ ©Úargsr$ r$ r% Únot_none¬ s rE )r. c G s t dd | D S )z?
|
Returns a boolean indicating if any argument is None.
|
c s s | ]}|d kV qd S r> r$ r? r$ r$ r% rB · s zany_none.<locals>.<genexpr>©r6 rC r$ r$ r% Úany_none³ s rG c G s t dd | D S )zA
|
Returns a boolean indicating if all arguments are None.
|
c s s | ]}|d kV qd S r> r$ r? r$ r$ r% rB ¾ s zall_none.<locals>.<genexpr>©ÚallrC r$ r$ r% Úall_noneº s rJ c G s t dd | D S )zC
|
Returns a boolean indicating if any argument is not None.
|
c s s | ]}|d k V qd S r> r$ r? r$ r$ r% rB Å s z any_not_none.<locals>.<genexpr>rF rC r$ r$ r% Ú any_not_noneÁ s rK c G s t dd | D S )zE
|
Returns a boolean indicating if all arguments are not None.
|
c s s | ]}|d k V qd S r> r$ r? r$ r$ r% rB Ì s z all_not_none.<locals>.<genexpr>rH rC r$ r$ r% Ú all_not_noneÈ s rL Úintc G s t dd | D S )z;
|
Returns the count of arguments that are not None.
|
c s s | ]}|d k V qd S r> r$ ©r@ Úxr$ r$ r% rB Ó s z!count_not_none.<locals>.<genexpr>)ÚsumrC r$ r$ r% Úcount_not_noneÏ s rQ .z ArrayLike | list | tuple | zipzNpDtype | Nonez
|
np.ndarray)Úvaluesr3 r. c C s d S r> r$ ©rR r3 r$ r$ r% Úasarray_tuplesafeÖ s rT r r c C s d S r> r$ rS r$ r$ r% rT à s c C sä t | ttfs"t| ds"t| } nt | tr2| jS t | trR|tjtfkrRt | S z4t
|
¡ " t
|
dtj ¡ tj| |d}W 5 Q R X W n tk
|
r¢ t | Y S X t|jjtrÀtj| td}|jdkràdd | D } t | }|S )NZ __array__Úignore©r3 é c S s g | ] }t |qS r$ )ÚtuplerN r$ r$ r% Ú
|
<listcomp>ÿ s z%asarray_tuplesafe.<locals>.<listcomp>)r0 r7 rX Úhasattrr Z_valuesr1 r4 Úobjectr ÚwarningsÚcatch_warningsÚ simplefilterZVisibleDeprecationWarningr5 r( Ú
|
issubclassr3 r9 ÚstrÚndim)rR r3 Úresultr$ r$ r% rT å s$
|
|
|
|
znp.ndarray | Iterable)Úlabelsr3 r. c C s\ t | ttfr| g} t | ttjfsLz t| } W n tk
|
rJ | g} Y nX t| |d} | S )zë
|
Transform label or iterable of labels to array, for use in Index.
|
|
Parameters
|
----------
|
dtype : dtype
|
If specified, use as dtype of the resulting array, otherwise infer.
|
|
Returns
|
-------
|
array
|
rV )r0 r` rX r7 r1 r2 Ú TypeErrorrT )rc r3 r$ r$ r% Úindex_labels_to_array s re c C s | d k r t | ttfs | gS | S r> )r0 rX r7 ©r* r$ r$ r% Úmaybe_make_list" s rg zIterable[T] | TzCollection[T] | T)r* r. c C s. t | tjr t | tjs t| S tt| } | S )zB
|
If obj is Iterable but not list-like, consume into list.
|
)r0 r r ÚSizedr7 r r rf r$ r$ r% Úmaybe_iterable_to_list( s
|
ri c C s( t | to&| jdko&| jdko&| jdkS )z
|
We have a null slice.
|
N©r0 ÚsliceÚstartÚstopÚsteprf r$ r$ r% Ú is_null_slice2 s
|
ÿþüro c C s* t | to(| jdk o(| jdk o(| j| jkS )z>
|
We have an empty slice, e.g. no values are selected.
|
N)r0 rk rl rm rf r$ r$ r% Úis_empty_slice> s
|
ÿþ
|
ürp z
|
list[bool]c C s dd | D S )zX
|
Find non-trivial slices in "line": return a list of booleans with same length.
|
c S s g | ]}t |tot| qS r$ )r0 rk ro )r@ Úkr$ r$ r% rY N s z"is_true_slices.<locals>.<listcomp>r$ )r" r$ r$ r% Úis_true_slicesJ s rr )r" r. c C s( t | to&| jdko&| j|ko&| jdkS )z&
|
We have a full length slice.
|
r Nrj )r* r" r$ r$ r% Ú is_full_sliceR s
|
ÿþürs c C s> t | drt| dS t| tr(t| jS t| r:t| jS d S )NÚ__name__) rZ Úgetattrr0 r Úget_callable_nameÚfuncÚcallabler9 rt rf r$ r$ r% rv ^ s
|
|
|
|
|
rv c K s t | r| |f|S | S )zØ
|
Evaluate possibly callable input using obj and kwargs if it is callable,
|
otherwise return as it is.
|
|
Parameters
|
----------
|
maybe_callable : possibly a callable
|
obj : NDFrame
|
**kwargs
|
)rx )Zmaybe_callabler* Úkwargsr$ r$ r% Úapply_if_callableo s rz c C sV t | ¡s(t| tr tt| jS t| } t| tj sBt
|
d| | tkrRt
|
d| S )a
|
Helper function to standardize a supplied mapping.
|
|
Parameters
|
----------
|
into : instance or subclass of collections.abc.Mapping
|
Must be a class, an initialized collections.defaultdict,
|
or an instance of a collections.abc.Mapping subclass.
|
|
Returns
|
-------
|
mapping : a collections.abc.Mapping subclass or other constructor
|
a callable object that can accept an iterator to create
|
the desired Mapping.
|
|
See Also
|
--------
|
DataFrame.to_dict
|
Series.to_dict
|
zunsupported type: z/to_dict() only accepts initialized defaultdicts) ÚinspectÚisclassr0 r r Údefault_factoryr9 r_ r ÚMappingrd )Zintor$ r$ r% Ústandardize_mapping s
|
|
r znp.random.Generator)Ústater. c C s d S r> r$ ©r r$ r$ r% Ú random_state s r zGint | ArrayLike | np.random.BitGenerator | np.random.RandomState | Noneznp.random.RandomStatec C s d S r> r$ r r$ r$ r% r ¥ s zRandomState | Noner c C sh t | s t| s t| tjjr*tj | ¡S t| tjjr<| S t| tjjrN| S | dkr\tjS tddS )aþ
|
Helper function for processing random_state arguments.
|
|
Parameters
|
----------
|
state : int, array-like, BitGenerator, Generator, np.random.RandomState, None.
|
If receives an int, array-like, or BitGenerator, passes to
|
np.random.RandomState() as seed.
|
If receives an np.random RandomState or Generator, just returns that unchanged.
|
If receives `None`, returns np.random.
|
If receives anything else, raises an informative ValueError.
|
|
.. versionchanged:: 1.1.0
|
|
array-like and BitGenerator object now passed to np.random.RandomState()
|
as seed
|
|
Default None.
|
|
Returns
|
-------
|
np.random.RandomState or np.random.Generator. If state is None, returns np.random
|
|
Nzdrandom_state must be an integer, array-like, a BitGenerator, Generator, a numpy RandomState, or None) r r r0 r1 ÚrandomZ BitGeneratorr r
|
r( r r$ r$ r% r ¬ s ÿþ ý ÿz/Callable[..., T] | tuple[Callable[..., T], str]r )rw r. c O sR t |tr>|\}}||kr,| d}t|| ||< |||S || f||S dS )as
|
Apply a function ``func`` to object ``obj`` either by passing obj as the
|
first argument to the function or, in the case that the func is a tuple,
|
interpret the first element of the tuple as a function and pass the obj to
|
that function as a keyword argument whose key is the value of the second
|
element of the tuple.
|
|
Parameters
|
----------
|
func : callable or tuple of (callable, str)
|
Function to apply to this object or, alternatively, a
|
``(callable, data_keyword)`` tuple where ``data_keyword`` is a
|
string indicating the keyword of ``callable`` that expects the
|
object.
|
*args : iterable, optional
|
Positional arguments passed into ``func``.
|
**kwargs : dict, optional
|
A dictionary of keyword arguments passed into ``func``.
|
|
Returns
|
-------
|
object : the return type of ``func``.
|
z/ is both the pipe target and a keyword argumentN)r0 rX r( )r* rw rD ry ÚtargetÚmsgr$ r$ r% Úpipeä s
|
|
|
r c s$ fdd}t tjtfr |S S )zv
|
Returns a function that will map names/labels, dependent if mapper
|
is a dict, Series or just a function.
|
c s | kr | S | S d S r> r$ )rO ©Úmapperr$ r% Úf s z get_rename_function.<locals>.f)r0 r r~ r )r r r$ r r% Úget_rename_function s r z"Hashable | Iterable | AnyArrayLikezlist | AnyArrayLike)rR r. c C s> t | ttjtttfr| S t | tjr8t | t s8t| S | gS )z¤
|
Convert list-like or scalar input to list-like. List, numpy and pandas array-like
|
inputs are returned unmodified whereas others are converted to list.
|
)
|
r0 r7 r1 r2 r r r r r r` )rR r$ r$ r% Úconvert_to_list_like s
|
r r` z Generator[(None, None, None)])Úattrr. c
|
c s4 t | |}t| || z
|
| V W 5 t| || X dS )zøTemporarily set attribute on an object.
|
|
Args:
|
obj: Object whose attribute will be modified.
|
attr: Attribute to modify.
|
value: Value to temporarily set attribute to.
|
|
Yields:
|
obj with modified attribute.
|
N)ru Úsetattr)r* r ÚvalueÚ old_valuer$ r$ r% Ú temp_setattr' s
|
|
|
r r ÚNone)Úindexr. c C s2 t | t |kr.tdt | dt | ddS )zC
|
Check the length of data matches the length of the index.
|
zLength of values (z") does not match length of index (ú)N)r8 r( )Údatar r$ r$ r% Úrequire_length_match; s ÿr rP ÚmaxÚminrI r6 ÚmeanÚprodÚstdÚvarÚmedianÚcumprodÚcumsumr z
|
str | None)rA r. c C s
|
t | ¡S )zH
|
if we define an internal function for this argument, return it
|
)Ú _cython_tableÚget©rA r$ r$ r% Úget_cython_funcn s r¢ c C s t | | ¡S )zd
|
if we define a builtin function for this argument, return it,
|
otherwise return the arg
|
)Ú_builtin_tabler r¡ r$ r$ r% Úis_builtin_funcu s r¤ zSequence[Hashable | None]zlist[Hashable])Únamesr. c C s dd t | D S )a,
|
If a name is missing then replace it by level_n, where n is the count
|
|
.. versionadded:: 1.4.0
|
|
Parameters
|
----------
|
names : list-like
|
list of column names or None values.
|
|
Returns
|
-------
|
list
|
list of column names with the None values replaced.
|
c S s& g | ] \}}|d kr d| n|qS )NZlevel_r$ )r@ Úir' r$ r$ r% rY s z&fill_missing_names.<locals>.<listcomp>)Ú enumerate)r¥ r$ r$ r% Úfill_missing_names} s r¨ ).).)N)N)N)lÚ__doc__Ú
|
__future__r ÚbuiltinsÚ collectionsr r Ú
|
contextlibÚ functoolsr r{ Útypingr r r r r
|
r r r r r r\ Únumpyr1 Z pandas._libsr Zpandas._typingr r r r r Zpandas.core.dtypes.castr Zpandas.core.dtypes.commonr r r r Zpandas.core.dtypes.genericr r r Z pandas.core.dtypes.inferencer Zpandas.core.dtypes.missingr Zpandasr r! r+ r: r= rE rG rJ rK rL rQ rT re rg ri ro rp rr rs rv rz r r r r r Úcontextmanagerr r rP r ÚmaximumÚreducer Zminimumr£ rI r6 Znansumr Znanmeanr Znanprodr Znanstdr Znanvarr Z nanmedianZnanmaxZnanminr Z
|
nancumprodr Z nancumsumr r¢ r¤ r¨ r$ r$ r$ r% Ú<module> sê 0 7 ÿ ! ÿ
|
8% ý ç
|
