o
    ņ©Zhū  ć                   @   sL   d dl mZ d dlmZmZ d dlZd dlZd dl	Z
d	ddZd
ddZdS )é    )ŚOrderedDict)ŚdateŚtimeNFc                 C   s  dt  ” fdt  ” fdt  ” fdt  ” fdt  ” fdt  ” fdt  ” fdt  ” fd	t  	” fd
t  
” fg
}t }g }|D ]'\}}| t  |t  |””” tjd|dtjd|ddtjd|dg||< q;| t  dt  t  ” ””” tjddgdddtjdgddtjg d¢ddg|d< | t  dt  t  d”””” tjg d¢ddddtjg d¢ddg|d< | ræ| t  dt  ” ”” t |”}t  |”}||fS )zå
    Dataframe with numpy arrays columns of every possible primitive type.

    Returns
    -------
    df: pandas.DataFrame
    schema: pyarrow.Schema
        Arrow schema definition that is in line with the constructed df.
    Śi1Śi2Śi4Śi8Śu1Śu2Zu4Śu8Zf4Zf8é
   ©Śdtypeé   Né   ŚstrŚ1õ   Ć¤Śobject©r   Ś2Ś3Z
datetime64Śms)z2007-07-13T01:23:34.123456789Nś2010-08-13T05:46:57.437699912zdatetime64[ms])z2007-07-13T02Nr   Ś__index_level_0__)ŚpaZint8Zint16Zint32Śint64Zuint8Zuint16Zuint32Zuint64Zfloat32Śfloat64r   ŚappendŚfieldŚlist_ŚnpZarangeŚstringŚarrayŚ	timestampŚpdŚ	DataFrameŚschema)Śinclude_indexZdtypesŚarraysŚfieldsr   Zarrow_dtypeŚdfr'   © r,   śT/var/www/html/lang_env/lib/python3.10/site-packages/pyarrow/tests/pandas_examples.pyŚdataframe_with_arrays   sH   
ü
üü
ż
żł

r.   c              
   C   s  t  }g }| t dt t ” ””” g d¢g d¢dg tjg d¢d tjdddd g|d< | t dt t ” ””” g d¢g d	¢dg t g d¢d ”ddd g|d< | t d
t t 	” ””” ddgddgg d¢g g|d
< | t dt t 
” ””” ddgddgg d¢g g|d< g tdddtdddgtdddgdtdddtdddgg}tdddtdddtdd d gg td!d"d gdtd#d#d#td$d#dtdddgg}t ” |ft ” |ft d%”|ft d&”|ft d'”|fg}|s|t d(”|fg7 }|D ]\}}d) |”}	t |”}
t |	|
”}| |” |||	< q| r2| t d*t ” ”” t |”}t |”}||fS )+a)  
    Dataframe with list columns of every possible primitive type.

    Returns
    -------
    df: pandas.DataFrame
    schema: pyarrow.Schema
        Arrow schema definition that is in line with the constructed df.
    parquet_compatible: bool
        Exclude types not supported by parquet
    r   )
r   r   é   é   é   r   é   é   é   é	   )r   r   r/   r0   r1   Nr/   r   Śdouble)
ē        ē      š?ē       @ē      @ē      @g      @g      @g      @g       @g      "@)r7   r8   r9   r:   r;   Z
bytes_listó   1ó   f)r<   ó   2ó   3Zstr_listr   r   r   iā  r   iš  é   é   iŠ  r2   r3   i±  r5   i“  r0   é   é   é;   é   r   r   é   Śsr   ŚusŚnsz{}_listr   )r   r   r   r   r    r   r!   r#   r   Śbinaryr"   r   r   Zdate32Zdate64Ztime32Ztime64Śformatr%   r&   r'   )r(   Zparquet_compatibler)   r*   Z	date_dataZ	time_dataZtemporal_pairsZ
value_typeŚdataŚ
field_nameZ
field_typer   r+   r'   r,   r,   r-   Śdataframe_with_listsS   s   ’’ūūūū	ū  ū
	
ū’




rN   )F)FF)Ścollectionsr   Śdatetimer   r   Śnumpyr!   Zpandasr%   Zpyarrowr   r.   rN   r,   r,   r,   r-   Ś<module>   s   
9