o h@sLddlmZddlmZmZddlZddlZddl Z d ddZ d ddZ dS) ) OrderedDict)datetimeNFc Csdtfdtfdtfdtfdtfdtfdtfdtfd t fd t fg }t }g}|D]'\}}| t |t|tjd |d tjd |d dtjd|d g||<q;| t dtttjddgdd dtjdgdd tjgddd g|d<| t dttdtjgddd ddtjgddd g|d<|r| t dtt|}t|}||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_dtypedfr4r;Q/var/www/vscode/kcb/lib/python3.10/site-packages/pyarrow/tests/pandas_examples.pydataframe_with_arrayssH           r=c Cst}g}|tdttgdgddgtjgddtjddddg|d<|tdttgdgd dgtgdddddg|d<|td tt d d gdd ggd gg|d <|tdtt ddgddggdgg|d<gt dddt dddgt dddgdt dddt dddgg}t dddt dddt dd d ggt d!d"d gdt d#d#d#t d$d#dt dddgg}t |ft|ftd%|ftd&|ftd'|fg}|s|td(|fg7}|D]\}}d)|} t|} t| | } || ||| <q|r2|td*tt|} t|} | | 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 )rrr>r?r@Nr>rdouble) ?@@@g@g@g@g @g"@)rFrGrHrIrJ bytes_list1f)rL23str_listrrriri irArBirDir? ;rrsrusnsz{}_listr)rr*rr+r,r#r-r0r)binaryr/rrdate32date64time32time64formatr2r3r4)r5parquet_compatibler7r8 date_data time_datatemporal_pairs value_typedata field_name field_typer+r:r4r;r;r<dataframe_with_listsSs                   ri)F)FF) collectionsrdatetimerrnumpyr-pandasr2pyarrowrr=rir;r;r;r<s  9