diff --git a/doc/source/user_guide/io.rst b/doc/source/user_guide/io.rst index 34c469bfc535b..eb0a9c6f60efb 100644 --- a/doc/source/user_guide/io.rst +++ b/doc/source/user_guide/io.rst @@ -5280,24 +5280,21 @@ Write to a parquet file. .. ipython:: python - df.to_parquet("example_pa.parquet", engine="pyarrow") - df.to_parquet("example_fp.parquet", engine="fastparquet") + # specify engine="pyarrow" or engine="fastparquet" to use a specific engine + df.to_parquet("example.parquet") Read from a parquet file. .. ipython:: python - result = pd.read_parquet("example_fp.parquet", engine="fastparquet") - result = pd.read_parquet("example_pa.parquet", engine="pyarrow") - + result = pd.read_parquet("example.parquet") result.dtypes By setting the ``dtype_backend`` argument you can control the default dtypes used for the resulting DataFrame. .. ipython:: python - result = pd.read_parquet("example_pa.parquet", engine="pyarrow", dtype_backend="pyarrow") - + result = pd.read_parquet("example.parquet", dtype_backend="pyarrow") result.dtypes .. note:: @@ -5309,24 +5306,14 @@ Read only certain columns of a parquet file. .. ipython:: python - result = pd.read_parquet( - "example_fp.parquet", - engine="fastparquet", - columns=["a", "b"], - ) - result = pd.read_parquet( - "example_pa.parquet", - engine="pyarrow", - columns=["a", "b"], - ) + result = pd.read_parquet("example.parquet", columns=["a", "b"]) result.dtypes .. ipython:: python :suppress: - os.remove("example_pa.parquet") - os.remove("example_fp.parquet") + os.remove("example.parquet") Handling indexes