U
|
¬ý°d¾ ã @ sb d dl mZ d dlZd dlZd dlZd dlmZ d dlm Z dddddd Z
|
d
|
d d d dZ dS )é )Ú annotationsN)ÚPerformanceWarning)Úfind_stack_levelz
|
str | NoneÚNone)ÚversionÚreturnc C s2 d}| dk r |d| d7 }t j|tt d dS )z`
|
Raise a PerformanceWarning for falling back to ExtensionArray's
|
non-pyarrow method
|
zGFalling back on a non-pyarrow code path which may decrease performance.Nz Upgrade to pyarrow >=z# to possibly suppress this warning.)Ú
|
stacklevel)ÚwarningsÚwarnr r )r Úmsg© r ú\d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/core/arrays/arrow/_arrow_utils.pyÚfallback_performancewarning s r znp.dtypez tuple[np.ndarray, np.ndarray])Údtyper c C s¦ t |¡}| ¡ }| j|j }t| |j }|d |||
}t j||d}|d }|dk rtjj t
|
¡ t| d|g| jd}t |¡}nt j t| t d}||fS )aµ
|
Convert a primitive pyarrow.Array to a numpy array and boolean mask based
|
on the buffers of the Array.
|
|
At the moment pyarrow.BooleanArray is not supported.
|
|
Parameters
|
----------
|
arr : pyarrow.Array
|
dtype : numpy.dtype
|
|
Returns
|
-------
|
(data, mask)
|
Tuple of two numpy arrays with the raw data (with specified dtype) and
|
a boolean mask (validity mask, so False means missing)
|
é )r r N)Úoffset)Únpr Úbuffersr ÚitemsizeÚlenZ
|
frombufferÚpyarrowZ BooleanArrayZ from_buffersZbool_ZasarrayZonesÚbool) Zarrr Zbuflistr ÚlengthZdata_bufÚdataZbitmaskÚmaskr r r Ú pyarrow_array_to_numpy_and_mask s"
|
ÿ r )N) Ú
|
__future__r r Únumpyr r Z pandas.errorsr Zpandas.util._exceptionsr r r r r r r Ú<module> s
|
