polars.from_arrow#
- polars.from_arrow(
- data: pa.Table | pa.Array | pa.ChunkedArray | pa.RecordBatch | Sequence[pa.RecordBatch],
- schema: SchemaDefinition | None = None,
- *,
- schema_overrides: SchemaDict | None = None,
- rechunk: bool = True,
Create a DataFrame or Series from an Arrow Table or Array.
This operation will be zero copy for the most part. Types that are not supported by Polars may be cast to the closest supported type.
- Parameters:
- data
pyarrow.Table
,pyarrow.Array
, one or morepyarrow.RecordBatch
Data representing an Arrow Table, Array, or sequence of RecordBatches.
- schemaSequence of str, (str,DataType) pairs, or a {str:DataType,} dict
The DataFrame schema may be declared in several ways:
As a dict of {name:type} pairs; if type is None, it will be auto-inferred.
As a list of column names; in this case types are automatically inferred.
As a list of (name,type) pairs; this is equivalent to the dictionary form.
If you supply a list of column names that does not match the names in the underlying data, the names given here will overwrite them. The number of names given in the schema should match the underlying data dimensions.
- schema_overridesdict, default None
Support type specification or override of one or more columns; note that any dtypes inferred from the schema param will be overridden.
- rechunkbool, default True
Make sure that all data is in contiguous memory.
- data
- Returns:
- DataFrame or Series
Examples
Constructing a DataFrame from an Arrow Table:
>>> import pyarrow as pa >>> data = pa.table({"a": [1, 2, 3], "b": [4, 5, 6]}) >>> df = pl.from_arrow(data) >>> df shape: (3, 2) ┌─────┬─────┐ │ a ┆ b │ │ --- ┆ --- │ │ i64 ┆ i64 │ ╞═════╪═════╡ │ 1 ┆ 4 │ │ 2 ┆ 5 │ │ 3 ┆ 6 │ └─────┴─────┘
Constructing a Series from an Arrow Array:
>>> import pyarrow as pa >>> data = pa.array([1, 2, 3]) >>> series = pl.from_arrow(data, schema={"s": pl.Int32}) >>> series shape: (3,) Series: 's' [i32] [ 1 2 3 ]