聽(tīng)說(shuō)Python也有類(lèi)似C++ STL的iterator,今天看了看,Python的東西真是好懂啊,幾分鐘就看完了iterator和genrator的Tutorial。Tutorial一開(kāi)始先提醒我們for語(yǔ)句是很強大滴,只要是個(gè)容器(container),用for就可以遍歷里面所有元素。例如:
for element in [1, 2, 3]:
print element
for element in (1, 2, 3):
print element
for key in {'one':1, 'two':2}:
print key
for char in "123":
print char
for line in open("myfile.txt"):
print line
然后Tutorial中接著(zhù)解釋?zhuān)鋵?shí)for是使用了in后面那個(gè)對象的iterator。每次調用iterator的next方法都可以得到下一個(gè)元素。到了最后一個(gè)元素時(shí)系統會(huì )拋出一個(gè)稱(chēng)為StopIteration的Exception,使for語(yǔ)句終止。只要為自己寫(xiě)的類(lèi)加上__iter__()方法和next()方法,也能夠使用for語(yǔ)句來(lái)遍歷所有元素。具體細節就不講了,因為這只是引出generator的一個(gè)過(guò)渡。
為了使iterator更直觀(guān)、生成更簡(jiǎn)單,Python引入了generator。generator對象可以起到類(lèi)似container的作用,container中的元素內容由generator中的語(yǔ)句確定。例如:
def reverse(data):
for index in range(len(data)-1, -1, -1):
yield data[index]
>>> for char in reverse('golf'):
... print char
...
f
l
o
g
在實(shí)現上,generator并不是生成一個(gè)列表,然后再由for對元素一個(gè)一個(gè)進(jìn)行處理,而是一次只返回一個(gè)元素(用yield語(yǔ)句)并保存generator的狀態(tài),等下次被調用時(shí)再從當前狀態(tài)往下執行,這樣可以省卻保存整個(gè)大列表的存儲代價(jià)。
為了練手,我寫(xiě)了兩個(gè)等價(jià)的程序,它們可以列出某個(gè)目錄下能找到的所有文件名:
(1)不用generator:
import os
def PrintName(fn):
print 'Found new file:' + fn
def ForEachFile(root, func):
for dir_entry in os.walk(root): # 對找到的每個(gè)目錄,os.walk()返回[目錄名,子目錄列表,文件名列表] (其實(shí)os.walk本身就是個(gè)generator)
for fName in dir_entry[2]:
fFullName = dir_entry[0]+'\'+fName
func(fFullName)
ForEachFile('d:\temp', PrintName)
(2)使用generator:
import os
def RecursiveFileList(root):
for dir_entry in os.walk(root):
for fName in dir_entry[2]:
yield dir_entry[0] + '\' + fName
for fn in RecursiveFileList('d:\temp'):
print 'Found new file: ', fn
對比之下,用了generator的封裝更加易懂和易用。
本站僅提供存儲服務(wù),所有內容均由用戶(hù)發(fā)布,如發(fā)現有害或侵權內容,請
點(diǎn)擊舉報。