## 問題 你在代碼中使用while循環來迭代處理數據，因為它需要調用某個函數或者和一般迭代模式不同的測試條件。能不能用迭代器來重寫這個循環呢？ ## 解決方案 一個常見的IO操作程序可能會想下面這樣： CHUNKSIZE = 8192 def reader(s): while True: data = s.recv(CHUNKSIZE) if data == b'': break process_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參數。