迭代器代替 while 無限循環

問題

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

解法

一個常見的 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 函數一個鮮為人知的特性是它接受一個可選的 callable 對象和一個標記 (結尾) 值作為輸入參數。當以這種方式使用的時候，它會創建一個迭代器，這個迭代器會不斷調用 callable 對象直到返回值和標記值相等為止。

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

總結

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

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