Searching in python/pyarrow