U
|
ý°dU ã @ sÜ d Z ddlmZ ddlmZmZ ddlZddlZddlZddl Z ddl
|
Z
|
ddl m Z m Z ddlZddlmZ ddlmZ ddlmZmZmZ ddlZdd lmZm Z dd
|
l m Z G d d d e Z G d d deZ dd Z!dS )a¦
|
Test extension array for storing nested data in a pandas container.
|
|
The JSONArray stores lists of dictionaries. The storage mechanism is a list,
|
not an ndarray.
|
|
Note
|
----
|
We currently store lists of UserDicts. Pandas has a few places
|
internally that specifically check for dicts, and does non-scalar things
|
in that case. We *want* the dictionaries to be treated as scalars, so we
|
hack around pandas by using UserDicts.
|
é )Ú annotations)ÚUserDictÚabcN)ÚAnyÚMapping)Útype_t©Ú'construct_1d_object_array_from_listlike)Ú is_bool_dtypeÚ is_list_likeÚ pandas_dtype)ÚExtensionArrayÚExtensionDtype)Úunpack_tuple_and_ellipsesc @ s8 e Zd ZU ejZdZe Zde d< e
|
ddddZ dS ) Ú JSONDtypeÚjsonzMapping[str, Any]Úna_valueztype_t[JSONArray]©Úreturnc C s t S )zq
|
Return the array type associated with this dtype.
|
|
Returns
|
-------
|
type
|
)Ú JSONArray)Úcls© r úXd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/extension/json/array.pyÚconstruct_array_type6 s z JSONDtype.construct_array_typeN) Ú__name__Ú
|
__module__Ú __qualname__r r ÚtypeÚnamer r Ú__annotations__Ú classmethodr r r r r r 1 s
|
|
r c @ sÒ e Zd Ze ZdZd,ddddZed-dd Zed
|
d Z d d Z
|
dd Z ddddZ dd Z dd Zd.ddZeddddZdd Zd/d d Zd d Zd0d"d#Zd$d% Zed&d' Zd(d) Zd*d+ ZdS )1r iè NFÚNoner c C sD |D ]&}t || jjstdt| jj q|| _| j | _| _d S )NzAll values must be of type )Ú
|
isinstanceÚdtyper Ú TypeErrorÚstrÚdataZ_itemsÚ_data)ÚselfÚvaluesr# ÚcopyÚvalr r r Ú__init__F s
|
zJSONArray.__init__c C s | |S ©Nr )r Zscalarsr# r* r r r Ú_from_sequenceS s zJSONArray._from_sequencec C s | dd |D S )Nc S s g | ]}|d krt |qS )r )r ©Ú.0Úxr r r Ú
|
<listcomp>Y s z.JSONArray._from_factorized.<locals>.<listcomp>r )r r) Úoriginalr r r Ú_from_factorizedW s zJSONArray._from_factorizedc sÌ t |trt|}t |tjr( j| S t |trL|td krLt jS t |trht j| S t|szt dnNt
|
j j |¡}t|jr® dd t |D ¡S t fdd|D S d S )Nzuonly integers, slices (`:`), ellipsis (`...`), numpy.newaxis (`None`) and integer or boolean arrays are valid indicesc S s g | ]\}}|r|qS r r )r0 r1 Úmr r r r2 q s z)JSONArray.__getitem__.<locals>.<listcomp>c s g | ]} j | qS r ©r& )r0 Úi©r( r r r2 s s )r" Útupler ÚnumbersÚIntegralr& Úslicer r Ú
|
IndexErrorÚpdÚapiZindexersZcheck_array_indexerr
|
r# r. Úzip)r( Úitemr r8 r Ú __getitem__[ s
|
|
|
ÿ
|
zJSONArray.__getitem__c C sÀ t |tjr|| j|< n¤t |t| tjfs8t |g¡}t |t j
|
r|j dkrt t ||D ],\}\}}|r\t || j js~t|| j|< q\n0t ||D ]$\}}t || j js°t|| j|< qd S )NÚbool)r" r: r; r& r r ÚSequenceÚ itertoolsÚcycleÚnpZndarrayr# Ú enumerater@ ÚAssertionError)r( ÚkeyÚvaluer7 ÚkÚvr r r Ú __setitem__u s zJSONArray.__setitem__Úintc C s
|
t | jS r- )Úlenr& r8 r r r Ú__len__ s zJSONArray.__len__c C s t S r- ©ÚNotImplemented©r( Úotherr r r Ú__eq__ s zJSONArray.__eq__c C s t S r- rR rT r r r Ú__ne__ s zJSONArray.__ne__c C s0 |d kr t }|t kr tt| S tj| j|dS )N©r# )Úobjectr ÚlistrG Úasarrayr& )r( r# r r r Ú __array__ s
|
zJSONArray.__array__c C s t | j¡S r- )ÚsysÚ getsizeofr& r8 r r r Únbytes s zJSONArray.nbytesc s t j fdd jD tdS )Nc s g | ]}| j jkqS r )r# r r/ r8 r r r2 s z"JSONArray.isna.<locals>.<listcomp>rX )rG Úarrayr& rC r8 r r8 r Úisna s zJSONArray.isnac
|
sÈ t |¡}d}|rz d kr"jj |dk ¡ r2tz fdd|D }W q¾ tk
|
rv } zt||W 5 d }~X Y q¾X nDzfdd|D }W n, tk
|
r¼ } zt||W 5 d }~X Y nX |¡S )NzIIndex is out of bounds or cannot do a non-empty take from an empty array.éÿÿÿÿc s" g | ]}|d krj | n qS )rb r6 ©r0 Úloc©Ú
|
fill_valuer( r r r2 ± s z"JSONArray.take.<locals>.<listcomp>c s g | ]} j | qS r r6 rc r8 r r r2 ¸ s )rG r[ r# r ÚanyÚ
|
ValueErrorr= r. )r( ZindexerZ
|
allow_fillrf ÚmsgÚoutputÚerrr re r Útake s&
|
ÿ ÿ
|
zJSONArray.takec C s t | | jd d
S r- ©r r& r8 r r r r* ¾ s zJSONArray.copyTc C s~ ddl m} t|}t|t| jr>|| jkr>|r:| ¡ S | S t||rd| t¡}| ¡ j
|
|ddS t j dd | D ||dS )Nr )Ú StringDtypeF)r* c S s g | ] }t |qS r ©Údictr/ r r r r2 Ñ s z$JSONArray.astype.<locals>.<listcomp>)r# r* ) Zpandas.core.arrays.string_rn r r" r r# r* Úastyper% r r. rG r` )r( r# r* rn rK r r r rq Á s
|
|
zJSONArray.astypec C s" t | dd dd | jD D S )Nc S s g | ] }t |qS r ro r/ r r r r2 Ö s z$JSONArray.unique.<locals>.<listcomp>c S s h | ]}t | ¡ qS r ©r9 Úitems)r0 Údr r r Ú <setcomp>Ö s z#JSONArray.unique.<locals>.<setcomp>rm r8 r r r ÚuniqueÓ s zJSONArray.uniquec C s" t tj dd |D ¡}| |S )Nc s s | ] }|j V qd S r- r6 r/ r r r Ú <genexpr>Ú s z.JSONArray._concat_same_type.<locals>.<genexpr>)rZ rE ÚchainÚ from_iterable)r Z to_concatr& r r r Ú_concat_same_typeØ s zJSONArray._concat_same_typec C s$ | ¡ }t|dkr | ¡ }|dfS )Nr r )Ú_values_for_argsortrP Zravel©r( Úfrozenr r r Ú_values_for_factorizeÝ s z JSONArray._values_for_factorizec C s dd | D }t |S )Nc S s g | ]}t | ¡ qS r rr r/ r r r r2 æ s z1JSONArray._values_for_argsort.<locals>.<listcomp>r r| r r r r{ ä s z JSONArray._values_for_argsort)NF)NF)N)FN)T)r r r r r# Z__array_priority__r, r r. r4 rB rN rQ rV rW r\ Úpropertyr_ ra rl r* rq rv rz r~ r{ r r r r r B s0
|
|
|
|
|
r c C s dd t dD S )Nc S s* g | ]"}t d d tt dd¡D qS )c S s$ g | ] }t tj¡t d d¡fqS )r éd )ÚrandomÚchoiceÚstringÚ ascii_lettersÚrandint©r0 Ú_r r r r2 î s ÿz(make_data.<locals>.<listcomp>.<listcomp>r é
|
)r Úranger r
r r r r r2 ì s úþÿz make_data.<locals>.<listcomp>r )r r r r r Ú make_dataê s ùr )"Ú__doc__Ú
|
__future__r Ú collectionsr r rE r: r r r] Útypingr r ÚnumpyrG Zpandas._typingr Zpandas.core.dtypes.castr Zpandas.core.dtypes.commonr
|
r r Zpandasr> Zpandas.api.extensionsr r Zpandas.core.indexersr r r r r r r r Ú<module> s& )
|
