Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/extension/arrow/__pycache__/arrays.cpython-37.pyc
Назад
B �5�g� � @ s� d Z ddlmZ ddlZddlZddlZddlZddlZ ddl m Z ddl Z ddlmZmZmZmZ ddlmZ ddlmZ eG dd � d e��ZeG d d � d e��ZG d d � d ee�ZG dd� de�ZG dd� de�ZdS )a Rudimentary Apache Arrow-backed ExtensionArray. At the moment, just a boolean array / type is implemented. Eventually, we'll want to parametrize the type and support multiple dtypes. Not all methods are implemented yet, and the current implementation is not efficient. � )� annotationsN)�type_t)�ExtensionArray�ExtensionDtype�register_extension_dtype�take)� is_scalar)�OpsMixinc @ sD e Zd ZejZdZdZej Z e dd�dd��Z e dd�dd ��Zd S ) �ArrowBoolDtype�bZ arrow_boolztype_t[ArrowBoolArray])�returnc C s t S )zq Return the array type associated with this dtype. Returns ------- type )�ArrowBoolArray)�cls� r �Y/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/extension/arrow/arrays.py�construct_array_type' s z#ArrowBoolDtype.construct_array_type�boolc C s dS )NTr )�selfr r r � _is_boolean2 s zArrowBoolDtype._is_booleanN)�__name__� __module__� __qualname__�np�bool_�type�kind�name�pa�NULL�na_value� classmethodr �propertyr r r r r r s r c @ s0 e Zd ZeZdZdZejZ e dd�dd��Z dS )�ArrowStringDtype�UZ arrow_stringztype_t[ArrowStringArray])r c C s t S )zq Return the array type associated with this dtype. Returns ------- type )�ArrowStringArray)r r r r r ? s z%ArrowStringDtype.construct_array_typeN) r r r �strr r r r r r r r r r r r r" 7 s r" c s� e Zd ZU ded< edd� �Zedd� �Zed3d d ��Zd d � Zd d� Z dd� Z d4� fdd� Z e dd� �Z dd� Zdd� Ze dd�d d ��Zd d � Zd5d d!�Zd"d#� Zed$d%� �Zd&d'� Zdd(�d)d*d+�d,d-�Zd6d/d0�Zd7d1d2�Z� ZS )8�ArrowExtensionArrayzpa.ChunkedArray�_datac C s t �t �t�|��g�}| |�S )N)r � chunked_array�arrayr Zasarray)r �values�arrr r r � from_scalarsN s z ArrowExtensionArray.from_scalarsc C s t |tj�st�| t�|g��S )N)� isinstancer ZArray�AssertionErrorr( )r r+ r r r � from_arrayS s z ArrowExtensionArray.from_arrayNFc C s | � |�S )N)r, )r Zscalars�dtype�copyr r r �_from_sequenceX s z"ArrowExtensionArray._from_sequencec C s t | �j� dt| j�� d�S )N�(�))r r �reprr' )r r r r �__repr__\ s z ArrowExtensionArray.__repr__c C s6 t |�r| j�� | S | j�� | }t| ��|�S d S )N)r r' � to_pandasr r, )r �item�valsr r r � __getitem___ s z ArrowExtensionArray.__getitem__c C s t | j�S )N)�lenr' )r r r r �__len__f s zArrowExtensionArray.__len__Tc s8 t |t| j��r*|| jkr*|r&| �� S | S t� �||�S )N)r- r r0 r1 �super�astype)r r0 r1 )� __class__r r r> i s zArrowExtensionArray.astypec C s | j S )N)�_dtype)r r r r r0 q s zArrowExtensionArray.dtypec C sT t |t| ��st� �|t�| j�t�|j��}tt�tj|t � | j� � �d�g��S )N)�mask) r- r �NotImplementedErrorr r) r' r r r( �pd�isnar7 )r �other�op�resultr r r �_logical_methodu s z#ArrowExtensionArray._logical_methodc C s t |t| ��sdS | �|tj�S )NF)r- r rH �operator�eq)r rE r r r �__eq__~ s zArrowExtensionArray.__eq__�int)r c C s t dd� | jjD ��S )Nc s s* | ]"}|� � D ]}|d k r|jV qqd S )N)Zbuffers�size)�.0�chunk�xr r r � <genexpr>� s z-ArrowExtensionArray.nbytes.<locals>.<genexpr>)�sumr' �chunks)r r r r �nbytes� s zArrowExtensionArray.nbytesc C s t �| j�� �}t| ��|�S )N)rC rD r' r7 r r, )r Znasr r r rD � s zArrowExtensionArray.isnac C s> | j �� }|r |d kr | jj}t||||d�}| j|| jd�S )N)� fill_value� allow_fill)r0 )r' r7 r0 r r r2 )r �indicesrV rU �datarG r r r r � s zArrowExtensionArray.takec C s t | �t�| j��S )N)r r1 r' )r r r r r1 � s zArrowExtensionArray.copyc C s, t tj�dd� |D ���}t�|�}| |�S )Nc s s | ]}|j jV qd S )N)r' rS )rN rP r r r rQ � s z8ArrowExtensionArray._concat_same_type.<locals>.<genexpr>)�list� itertools�chain� from_iterabler r( )r Z to_concatrS r+ r r r �_concat_same_type� s z%ArrowExtensionArray._concat_same_typec C s t | ��| j�� �S )N)r r, r' r7 )r r r r � __invert__� s z ArrowExtensionArray.__invert__)�skipnar% r )r r_ c K sZ |r| | � � }n| }yt||�}W n( tk rN } z t|�W d d }~X Y nX |f |�S )N)rD �getattr�AttributeError� TypeError)r r r_ �kwargsr+ rF �errr r r �_reduce� s zArrowExtensionArray._reducer c C s t | j�� �� �S )N)r r' r7 �any)r �axis�outr r r rf � s zArrowExtensionArray.anyc C s t | j�� �� �S )N)r r' r7 �all)r rg rh r r r ri � s zArrowExtensionArray.all)NF)T)FN)r N)r N)r r r �__annotations__r r, r/ r2 r6 r: r< r> r! r0 rH rK rT rD r r1 r] r^ re rf ri � __classcell__r r )r? r r& K s* r& c @ s e Zd Zdd� ZdS )r c C s4 t |tj�st�|jt�� ks"t�|| _t� | _ d S )N) r- r � ChunkedArray� ValueErrorr r r. r' r r@ )r r* r r r �__init__� s zArrowBoolArray.__init__N)r r r rn r r r r r � s r c @ s e Zd Zdd� ZdS )r$ c C s4 t |tj�st�|jt�� ks"t�|| _t� | _ d S )N) r- r rl rm r �stringr. r' r" r@ )r r* r r r rn � s zArrowStringArray.__init__N)r r r rn r r r r r$ � s r$ ) �__doc__� __future__r r1 rZ rI �numpyr Zpyarrowr Zpandas._typingr ZpandasrC Zpandas.api.extensionsr r r r Zpandas.api.typesr Zpandas.core.arrayliker r r" r& r r$ r r r r �<module> s$ p
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.05 |
proxy
|
phpinfo
|
Настройка