問題描述

使用文本文件，我可以這樣寫:

With a text file, I can write this:

with open(path, 'r') as file:
    for line in file:
        # handle the line

這等價于:

with open(path, 'r') as file:
    for line in iter(file.readline, ''):
        # handle the line

這個成語記錄在 PEP 234 但我沒有找到二進(jìn)制文件的類似習(xí)語.

This idiom is documented in PEP 234 but I have failed to locate a similar idiom for binary files.

使用二進(jìn)制文件，我可以這樣寫:

With a binary file, I can write this:

with open(path, 'rb') as file:
    while True:
        chunk = file.read(1024 * 64)
        if not chunk:
            break
        # handle the chunk

我嘗試過與文本文件相同的習(xí)語:

I have tried the same idiom that with a text file:

def make_read(file, size):
    def read():
        return file.read(size)
    return read

with open(path, 'rb') as file:
    for chunk in iter(make_read(file, 1024 * 64), b''):
        # handle the chunk

這是在 Python 中迭代二進(jìn)制文件的慣用方式嗎?

Is it the idiomatic way to iterate over a binary file in Python?

推薦答案

我不知道有什么內(nèi)置的方法可以做到這一點(diǎn)，但是一個包裝函數(shù)很容易寫:

I don't know of any built-in way to do this, but a wrapper function is easy enough to write:

def read_in_chunks(infile, chunk_size=1024*64):
    while True:
        chunk = infile.read(chunk_size)
        if chunk:
            yield chunk
        else:
            # The chunk was empty, which means we're at the end
            # of the file
            return

然后在交互式提示處:

>>> from chunks import read_in_chunks
>>> infile = open('quicklisp.lisp')
>>> for chunk in read_in_chunks(infile):
...     print chunk
... 
<contents of quicklisp.lisp in chunks>

當(dāng)然，您可以輕松地將其調(diào)整為使用 with 塊:

Of course, you can easily adapt this to use a with block:

with open('quicklisp.lisp') as infile:
    for chunk in read_in_chunks(infile):
        print chunk

你可以像這樣去掉 if 語句.

And you can eliminate the if statement like this.

def read_in_chunks(infile, chunk_size=1024*64):
    chunk = infile.read(chunk_size)
    while chunk:
        yield chunk
        chunk = infile.read(chunk_size)

這篇關(guān)于迭代二進(jìn)制文件的慣用方法是什么?的文章就介紹到這了，希望我們推薦的答案對大家有所幫助，也希望大家多多支持html5模板網(wǎng)！