
    z-Ph                     J    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ZddZdS )	    )OrderedDict)datetimeNFc                    dt          j                    fdt          j                    fdt          j                    fdt          j                    fdt          j                    fdt          j                    fdt          j                    fdt          j                    fd	t          j	                    fd
t          j
                    fg
}t                      }g }|D ]\  }}|                    t          j        |t          j        |                               t          j        d|          t          j        d|          dt          j        d|          g||<   |                    t          j        dt          j        t          j                                                   t          j        ddgd          dt          j        dgd          t          j        g dd          g|d<   |                    t          j        dt          j        t          j        d                                         t          j        g dd          ddt          j        g dd          g|d<   | r9|                    t          j        dt          j                                         t)          j        |          }t          j        |          }||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u2u4u8f4f8
   dtype   N   str1   äobjectr   23
datetime64ms)z2007-07-13T01:23:34.123456789N2010-08-13T05:46:57.437699912zdatetime64[ms])z2007-07-13T02Nr   __index_level_0__)paint8int16int32int64uint8uint16uint32uint64float32float64r   appendfieldlist_nparangestringarray	timestamppd	DataFrameschema)include_indexdtypesarraysfieldsr   arrow_dtypedfr6   s           ]/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/pyarrow/tests/pandas_examples.pydataframe_with_arraysr>      s    RWYY$

!3RXZZ 4"4RXZZ 4"5RY[[!D")++#6RZ\\"T2:<<$8	:F ]]FF$ 
 
{bhubh{&;&;<<===Ib&&&Iau%%%Iau%%%	
u MM"(5"(29;;"7"788999
#tH---
#h'''
111	F5M MM"(<",t2D2D)E)EFFGGG
 3 3 3 (	) 	) 	) 	
 3 3 3 (	) 	) 	)F<  Abh2BHJJ??@@@	f		BYvFv:    c                    t                      }g }|                    t          j        dt          j        t          j                                                   g dg ddg t          j        g ddz  t          j                  ddd         g|d<   |                    t          j        dt          j        t          j                                                   g dg d	dg t          j        g ddz            ddd         g|d<   |                    t          j        d
t          j        t          j	                                                   ddgddgg dg g|d
<   |                    t          j        dt          j        t          j
                                                   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          j                    |ft          j                    |ft          j        d%          |ft          j        d&          |ft          j        d'          |fg}|s|t          j        d(          |fgz  }|D ]]\  }}d)                    |          }	t          j        |          }
t          j        |	|
          }|                    |           |||	<   ^| r9|                    t          j        d*t          j                                         t%          j        |          }t          j        |          }||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   rA   rB   rC   NrA   r   double)
              ?       @      @      @g      @g      @g      @g       @g      "@)rI   rJ   rK   rL   rM   
bytes_list   1   f)rO      2   3str_listr   r   r   i  r   i        i  rD   rE   i  rG   i  rB         ;      r   r      sr   usnsz{}_listr    )r   r,   r!   r-   r.   r%   r/   r2   r+   binaryr1   r   r   date32date64time32time64formatr4   r5   r6   )r7   parquet_compatibler9   r:   	date_data	time_datatemporal_pairs
value_typedata
field_name
field_typer-   r<   r6   s                 r=   dataframe_with_listsrl   S   s    ]]FF
MM"(7BHRXZZ$8$899:::&&&

///!3x	! 	! 	!!$1	&F7O MM"(8RXbjll%;%;<<===000

999A=>>sssCF8 MM"(<")++)>)>??@@@	t	
F< MM"(:rx	'<'<==>>>	d	
F: 		dAq		4b"--.	dAq			dAq		4a++,I 
b"b		41a==$r2r*:*:;
	b!R	aARARA7I 
i 	i 	3#	4)$	4)$N  
Yt__i(
 	
 + " "
D%%j11
Xj))
Z00e!z Abh2BHJJ??@@@	f		BYvFv:r?   )F)FF)collectionsr   datetimer   r   numpyr/   pandasr4   pyarrowr!   r>   rl    r?   r=   <module>rs      s   $ $ # # # # #                    6 6 6 6rY Y Y Y Y Yr?   