K iHddlmZddlmZmZddlZddlZddl Z ddZ ddZ y)) OrderedDict)datetimeNc dtjfdtjfdtjfdtjfdtj fdtj fdtjfdtjfd tjfd tjfg }t}g}|D]\}}|jtj|tj|tj d | tj d | dtj d| g||<|jtjdtjtj"tj$ddgd dtj$dgd tj$gdd g|d<|jtjdtjtj&dtj$gdd ddtj$gdd g|d<|r7|jtjdtjt)j*|}tj,|}||fS)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. i1i2i4i8u1u2u4u8f4f8 dtypeNstr1äobjectr23 datetime64ms)z2007-07-13T01:23:34.123456789N2010-08-13T05:46:57.437699912zdatetime64[ms])z 2007-07-13T02Nr__index_level_0__)paint8int16int32int64uint8uint16uint32uint64float32float64rappendfieldlist_nparangestringarray timestamppd DataFrameschema) include_indexdtypesarraysfieldsr arrow_dtypedfr6s c/mnt/ssd/data/python-lab/Trading/venv/lib/python3.12/site-packages/pyarrow/tests/pandas_examples.pydataframe_with_arraysr>s&RWWY$ !3RXXZ 4"4RXXZ 4"5RYY[!D"))+#6RZZ\"T2::<$8 :F ]F F$ { bhhubhh{&;<= IIb & IIau %  IIau %  u   MM"((5"((299;"789 #tH-  #h' 1 F5M MM"((<",,t2D)EFG 3( )   3( ) F< bhh2BHHJ?@ f B YYv F v:c 6t}g}|jtjdtjtj gdgddgt jgddzt j dddg|d<|jtjdtjtjgdgd dgt jgddzdddg|d<|jtjd tjtjd d gdd ggd gg|d <|jtjdtjtjddgddggdgg|d<gtdddtdddgtdddgdtdddtdddgg}tdddtdddtdd d ggtd!d"d gdtd#d#d#td$d#dtdddgg}tj|ftj|ftjd%|ftjd&|ftj d'|fg}|s|tj d(|fgz }|D]K\}}|d)} tj|} tj| | } |j| ||| <M|r7|jtjd*tj t#j$|} tj&|} | | fS)+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%) rrr )rrrArBrCNrArdouble) ?@@@g@g@g@g @g"@)rIrJrKrLrM bytes_list1f)rO23str_listrrriri irDrEirGirB ;rrsrusns_listr )rr,r!r-r.r%r/r2r+binaryr1rrdate32date64time32time64r4r5r6)r7parquet_compatibler9r: date_data time_datatemporal_pairs value_typedata field_name field_typer-r<r6s r=dataframe_with_listsrlSs.]F F MM"((7BHHRXXZ$89:&   /!3xx !!$1 & F7O MM"((8RXXbjjl%;<=0   9A=>ssC F8 MM"((<"))+)>?@ t   F< MM"((:rxx '<=> d   F: dAq 4b"-. dAq  dAq 4a+, I b"b 41a=$r2r*:; b!R aARARA7 I i i 3# 4)$ 4)$ N  YYt_i (  +" D"|5) XXj) Z0 e!z " bhh2BHHJ?@ f B YYv F v:r?)F)FF) collectionsrdatetimerrnumpyr/pandasr4pyarrowr!r>rlr?r=rss"$$6rYr?