
    g                     H    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y)    )OrderedDict)datetimeNc                    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 ]  \  }}|j                  t        j                  |t        j                  |                   t        j                   d|      t        j                   d|      dt        j                   d|      g||<    |j                  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<   |j                  t        j                  dt        j                  t        j&                  d                         t        j$                  g dd      ddt        j$                  g dd      g|d<   | r7|j                  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           R/var/www/openai/venv/lib/python3.12/site-packages/pyarrow/tests/pandas_examples.pydataframe_with_arraysr>      s!    RWWY$
!3RXXZ 4"4RXXZ 4"5RYY[!D"))+#6RZZ\"T2::<$8	:F ]FF${bhhubhh{&;<=IIb&IIau%IIau%	
u % MM"((5"((299;"789
#tH-
#h'
1	F5M MM"((<",,t2D)EFG
 3 (	) 	
 3 (	)F< bhh2BHHJ?@	f	BYYvFv:    c                 N   t               }g }|j                  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<   |j                  t        j                  dt        j                  t        j                                            g dg d	dg t        j                  g ddz        ddd   g|d<   |j                  t        j                  d
t        j                  t        j                                            ddgddgg dg g|d
<   |j                  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 ]W  \  }}d)j#                  |      }	t        j                  |      }
t        j                  |	|
      }|j                  |       |||	<   Y | r7|j                  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"((7BHHRXXZ$89:&

/!3xx	!!$1	&F7O MM"((8RXXbjjl%;<=0

9A=>ssCF8 MM"((<"))+)>?@	t	
F< MM"((:rxx		'<=>	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 YYt_i(
 	
 +
D%%j1
XXj)
Z0e!z + bhh2BHHJ?@	f	BYYvFv:r?   )F)FF)collectionsr   datetimer   r   numpyr/   pandasr4   pyarrowr!   r>   rl    r?   r=   <module>rs      s"   $ $    6rYr?   