arro3_core_arrow_arrays_tables_pyarrow_interop.py

1import arro3.core as arrow
2import pyarrow as pa
3
4# Create an Array from a Python list
5# Equivalent to pa.array([1, 2, 3])
6array = arrow.Array([1, 2, 3])
7print(f"Arrow Array: {array}")
8
9# Create a ChunkedArray from a list of lists
10# Equivalent to pa.chunked_array([[1, 2], [3]])
11chunked_array = arrow.ChunkedArray([[1, 2], [3]])
12print(f"Chunked Array: {chunked_array}")
13
14# Create a Table from a dictionary of lists
15# Equivalent to pa.table({"a": [1, 2, 3]})
16table = arrow.Table.from_pydict({"a": [1, 2, 3]})
17print(f"Table: {table}")
18
19# Interop with pyarrow using the Arrow PyCapsule Interface
20pa_table = pa.table(table)
21print(f"PyArrow Table: {pa_table}")
22
23# Convert back from pyarrow
24arro3_table = arrow.Table(pa_table)
25print(f"Arro3 Table from PyArrow: {arro3_table}")