I looked at the standard documentation that I would expect to capture my need (Apache Arrow and Pandas), and I could not seem to figure it out.
I know Python best, so I would like to use Python, but it is not a strict requirement.
I need to move Parquet files from one location (a URL) to another (an Azure storage account, in this case using the Azure machine learning platform, but this is irrelevant to my problem).
These files are too large to simply perform pd.read_parquet("https://my-file-location.parquet")
, since this reads the whole thing into an object.
I thought that there must be a simple way to create a file object and stream that object line by line -- or maybe column chunk by column chunk. Something like
import pyarrow.parquet as pq
with pq.open("https://my-file-location.parquet") as read_file_handle:
with pq.open("https://my-azure-storage-account/my-file.parquet", "write") as write_filehandle:
for next_line in read_file_handle{
write_file_handle.append(next_line)
I understand it will be a little different because Parquet is primarily meant to be accessed in a columnar fashion. Maybe there is some sort of config object that I would pass which specifies which columns of interest, or maybe how many lines can be grabbed in a chunk or something similar.
But the key expectation is that there is a means to access a parquet file without loading it all into memory. How can I do this?
FWIW, I did try to just use Python's standard open
function, but I was not sure how to use open
with a URL location and a byte stream. If it is possible to do this via just open
and skip anything Parquet-specific, that is also fine.
Some of the comments have suggested using bash-like scripts, such as here. I can use this if there is nothing else, but it is not ideal because:
Great post, based on @Micah's answer, I put my 2 cents in it, in case you don't want to read the docs. A small snippet is the following:
import pandas as pd
import numpy as np
from pyarrow.parquet import ParquetFile
# create a random df then save to parquet
df = pd.DataFrame({
'A': np.arange(10000),
'B': np.arange(10000),
'C': np.arange(10000),
'D': np.arange(10000),
})
df.to_parquet('./test/test')
# ****** below is @Micah Kornfield's answer ******
# 1. open parquet file
batch = ParquetFile('./test/test')
# 2. define generator of batches
record = batch.iter_batches(
batch_size=10,
columns=['B', 'C'],
)
# 3. yield pandas/numpy data
print(next(record).to_pandas()) # pandas
print(next(record).to_pydict()) # native python dict