1 回答

TA貢獻1840條經驗 獲得超5個贊
問題在于gzip模塊會立即將整個文件解壓縮,然后將未壓縮的文件存儲在內存中。對于這么大的文件,更實際的方法是一次將文件解壓縮一點,流式傳輸結果。
#!/usr/bin/env python3
import io
import zlib
def stream_unzipped_bytes(filename):
"""
Generator function, reads gzip file `filename` and yields
uncompressed bytes.
This function answers your original question, how to read the file,
but its output is a generator of bytes so there's another function
below to stream these bytes as text, one line at a time.
"""
with open(filename, 'rb') as f:
wbits = zlib.MAX_WBITS | 16 # 16 requires gzip header/trailer
decompressor = zlib.decompressobj(wbits)
fbytes = f.read(16384)
while fbytes:
yield decompressor.decompress(decompressor.unconsumed_tail + fbytes)
fbytes = f.read(16384)
def stream_text_lines(gen):
"""
Generator wrapper function, `gen` is a bytes generator.
Yields one line of text at a time.
"""
try:
buf = next(gen)
while buf:
lines = buf.splitlines(keepends=True)
# yield all but the last line, because this may still be incomplete
# and waiting for more data from gen
for line in lines[:-1]:
yield line.decode()
# set buf to end of prior data, plus next from the generator.
# do this in two separate calls in case gen is done iterating,
# so the last output is not lost.
buf = lines[-1]
buf += next(gen)
except StopIteration:
# yield the final data
if buf:
yield buf.decode()
# Sample usage, using the stream_text_lines generator to stream
# one line of RDF text at a time
bytes_generator = (x for x in stream_unzipped_bytes('freebase-rdf-latest.gz'))
for line in stream_text_lines(bytes_generator):
# do something with `line` of text
print(line, end='')
添加回答
舉報