polars.read_parquet#

polars.read_parquet(source: str | Path | BinaryIO | BytesIO | bytes, *, columns: list[int] | list[str] | None = None, n_rows: int | None = None, use_pyarrow: bool = False, memory_map: bool = True, storage_options: dict[str, Any] | None = None, parallel: ParallelStrategy = 'auto', row_count_name: str | None = None, row_count_offset: int = 0, low_memory: bool = False, pyarrow_options: dict[str, Any] | None = None, use_statistics: bool = True, rechunk: bool = True) DataFrame[source]#

Read into a DataFrame from a parquet file.

Parameters:
source

Path to a file, or a file-like object. If the path is a directory, that directory will be used as partition aware scan. If fsspec is installed, it will be used to open remote files.

columns

Columns to select. Accepts a list of column indices (starting at zero) or a list of column names.

n_rows

Stop reading from parquet file after reading n_rows. Only valid when use_pyarrow=False.

use_pyarrow

Use pyarrow instead of the Rust native parquet reader. The pyarrow reader is more stable.

memory_map

Memory map underlying file. This will likely increase performance. Only used when use_pyarrow=True.

storage_options

Extra options that make sense for fsspec.open() or a particular storage connection, e.g. host, port, username, password, etc.

parallel{‘auto’, ‘columns’, ‘row_groups’, ‘none’}

This determines the direction of parallelism. ‘auto’ will try to determine the optimal direction.

row_count_name

If not None, this will insert a row count column with give name into the DataFrame.

row_count_offset

Offset to start the row_count column (only use if the name is set).

low_memory

Reduce memory pressure at the expense of performance.

pyarrow_options

Keyword arguments for pyarrow.parquet.read_table.

use_statistics

Use statistics in the parquet to determine if pages can be skipped from reading.

rechunk

Make sure that all columns are contiguous in memory by aggregating the chunks into a single array.

Returns:
DataFrame

Notes

This operation defaults to a rechunk operation at the end, meaning that all data will be stored continuously in memory. Set rechunk=False if you are benchmarking the parquet-reader. A rechunk is an expensive operation.