問題

你想在一個文件里每次讀入固定大小的字節，比如每次讀入4個字節并轉成int，或者每次讀入x個字節并進行結構化，例如：

l = [5, 2, 4, 1, 2, 4, 5, 6, 8]

with open('test', 'wb') as fd:

for num in l:

fd.write(num.to_bytes(4, 'big'))

解決方案

可以簡單的用while循環來完成

with open('test', 'rb') as fd:

r = fd.read(4)

while r:

print(int.from_bytes(r, 'big'), end=' ')

r = fd.read(4)

輸出為5 2 4 1 2 4 5 6 8

但更優雅的做法是結合使用iter和functools.partial

from functools import partial

with open('test', 'rb') as fd:

for r in iter(partial(fd.read, 4), b''):

print(int.from_bytes(r, 'big'), end=' ')

輸出同樣是5 2 4 1 2 4 5 6 8

討論

functools.partial的作用是對一個函數進行包裝（可以將一些參數的值固定）并生成一個新的簽名，例如：

>>> from functools import partial

>>> basetwo = partial(int, base=2)

>>> basetwo.__doc__ = 'Convert base 2 string to an int.'

>>> basetwo('10010')

18

functools.partial的大致實現如下：

def partial(func, *args, **keywords):

def newfunc(*fargs, **fkeywords):

newkeywords = keywords.copy()

newkeywords.update(fkeywords)

return func(*args, *fargs, **newkeywords)

newfunc.func = func

newfunc.args = args

newfunc.keywords = keywords

return newfunc

而iter的作用是當只有一個參數時，對這個對象進行迭代，所以這個對象必須實現__iter()__或__getitem()__方法，例如：

>>> l = [1, 2, 3, 4]

>>> for i in iter(l):

... print(i, end=' ')

...

>>> 1 2 3 4

但當有兩個參數時，第一個參數必須是可調用的（比如函數），第二個參數是終止的值，當調用第一個參數返回的結果等于第二個參數時，迭代就停止了，例如：

with open('mydata.txt') as fp:

for line in iter(fp.readline, ''):

process_line(line)

所以綜合兩個方法來看上面的iter(partial(fd.read, 4), b'')，就是每次調用fd.read(4)直到返回一個空的bytes

來源

Python Cookbook

關注

歡迎關注我的微信公眾號：python每日一練

總結

以上是生活随笔為你收集整理的python每日一练名片管理程序_Python每日一练0022的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。