Skip to content

Commit 9baf420

Browse files
add filters to read_pandas
1 parent 0df8c88 commit 9baf420

File tree

2 files changed

+5
-1
lines changed

2 files changed

+5
-1
lines changed

python/pyarrow/parquet.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -1215,10 +1215,11 @@ def read_table(source, columns=None, use_threads=True, metadata=None,
12151215

12161216

12171217
def read_pandas(source, columns=None, use_threads=True, memory_map=True,
1218-
metadata=None):
1218+
metadata=None, filters=None):
12191219
return read_table(source, columns=columns,
12201220
use_threads=use_threads,
12211221
metadata=metadata, memory_map=True,
1222+
filters=filters,
12221223
use_pandas_metadata=True)
12231224

12241225

python/pyarrow/tests/test_parquet.py

+3
Original file line numberDiff line numberDiff line change
@@ -1546,6 +1546,9 @@ def test_filters_read_table(tempdir):
15461546
base_path, filesystem=fs, filters=[('integers', '<', 3)])
15471547
assert table.num_rows == 3
15481548

1549+
table = pq.read_pandas(
1550+
base_path, filters=[('integers', '<', 3)])
1551+
assert table.num_rows == 3
15491552

15501553
@pytest.yield_fixture
15511554
def s3_example():

0 commit comments

Comments
 (0)