迭代器代替 while 無限循環(huán)

問題

你在代碼中使用 while 循環(huán)來迭代處理數(shù)據(jù)，因為它需要調(diào)用某個函數(shù)或者和一般迭代模式不同的測試條件。能不能用迭代器來重寫這個循環(huán)呢？

解法

一個常見的 IO 操作程序可能會想下面這樣：

CHUNKSIZE = 8192 def reader(s):while True:data = s.recv(CHUNKSIZE)if data == b'':breakprocess_data(data)

這種代碼通常可以使用 iter() 來代替，如下所示：

def reader2(s):for chunk in iter(lambda: s.recv(CHUNKSIZE), b''):pass# process_data(data)

如果你懷疑它到底能不能正常工作，可以試驗下一個簡單的例子。比如：

>>> import sys >>> f = open('/etc/passwd') >>> for chunk in iter(lambda: f.read(10), ''): ... n = sys.stdout.write(chunk) ... nobody:*:-2:-2:Unprivileged User:/var/empty:/usr/bin/false root:*:0:0:System Administrator:/var/root:/bin/sh daemon:*:1:1:System Services:/var/root:/usr/bin/false _uucp:*:4:4:Unix to Unix Copy Protocol:/var/spool/uucp:/usr/sbin/uucico ... >>>

討論

iter 函數(shù)一個鮮為人知的特性是它接受一個可選的 callable 對象和一個標記 (結尾) 值作為輸入?yún)?shù)。當以這種方式使用的時候，它會創(chuàng)建一個迭代器，這個迭代器會不斷調(diào)用 callable 對象直到返回值和標記值相等為止。

這種特殊的方法對于一些特定的會被重復調(diào)用的函數(shù)很有效果，比如涉及到 I/O調(diào)用的函數(shù)。舉例來講，如果你想從套接字或文件中以數(shù)據(jù)塊的方式讀取數(shù)據(jù)，通常你得要不斷重復的執(zhí)行 read() 或 recv() ，并在后面緊跟一個文件結尾測試來決定是否終止。這節(jié)中的方案使用一個簡單的 iter() 調(diào)用就可以將兩者結合起來了。其中l(wèi)ambda 函數(shù)參數(shù)是為了創(chuàng)建一個無參的 callable 對象，并為 recv 或 read() 方法提供了 size 參數(shù)。

總結

以上是生活随笔為你收集整理的《Python Cookbook 3rd》笔记（4.16）：迭代器代替 while 无限循环的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。