o R “ióć@sLddlmZddlmZmZddlZddlZddl Z d dd„Z d dd„Z dS) é)Ś OrderedDict)ŚdateŚtimeNFc Csšdt ”fdt ”fdt ”fdt ”fdt ”fdt ”fdt ”fdt ”fd t  ”fd t  ”fg }t ƒ}g}|D]'\}}|  t  |t |”””tjd |d tjd |d dtjd|d g||<q;|  t  dt t ””””tjddgdd dtjdgdd tjgd¢dd g|d<|  t  dt t d””””tjgd¢dd ddtjgd¢dd g|d<|ræ|  t  dt ”””t |”}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. Śi1Śi2Śi4Śi8Śu1Śu2Śu4Śu8Śf4Śf8é ©ŚdtypeéNéŚstrŚ1õƤŚobject©rŚ2Ś3Ś datetime64Śms)z2007-07-13T01:23:34.123456789Nś2010-08-13T05:46:57.437699912zdatetime64[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Śdfr4©r;śQ/home/ubuntu/.local/lib/python3.10/site-packages/pyarrow/tests/pandas_examples.pyŚdataframe_with_arrayssH ü     üü ż żł   r=c Cs€tƒ}g}| t dt t ””””gd¢gd¢dgtjgd¢dtjdddd…g|d<| t dt t ””””gd¢gd ¢dgt gd¢d”ddd…g|d<| t d t t  ””””d d gdd ggd ¢gg|d <| t dt t  ””””ddgddggd¢gg|d<gt dddƒt dddƒgt dddƒgdt dddƒt dddƒgg}t dddƒt dddƒt dd d ƒggt d!d"d ƒgdt d#d#d#ƒt d$d#dƒt dddƒgg}t  ”|ft ”|ft d%”|ft d&”|ft d'”|fg}|s|t d(”|fg7}|D]\}}|›d)} t |”} t | | ”} | | ”||| <q|r2| t d*t ”””t |”} 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#) rréééréééé )rrr>r?r@Nr>rŚdouble) ēēš?ē@ē@ē@g@g@g@g @g"@)rFrGrHrIrJŚ bytes_listó1óf)rLó2ó3Śstr_listrrriārišé éiŠrArBi±rDi“r?éé é;érréŚsrŚusŚnsŚ_listr)rr*rr+r,r#r-r0r)Śbinaryr/rrŚdate32Śdate64Śtime32Śtime64r2r3r4)r5Śparquet_compatibler7r8Ś date_dataŚ time_dataŚtemporal_pairsŚ value_typeŚdataŚ field_nameŚ field_typer+r:r4r;r;r<Śdataframe_with_listsSs† ’’ūūūū  ū   ū    ū ’        ri)F)FF) Ś collectionsrŚdatetimerrŚnumpyr-Śpandasr2Śpyarrowrr=rir;r;r;r<Śs  9