Python中檔案遍曆的兩種方法

關於Python的檔案遍曆，大概有兩種方法，一種是較為便利的os.walk()，還有一種是利用os.listdir()遞迴遍曆。

方法一：利用os.walk

os.walk可以自頂向下或者自底向上遍曆整個檔案樹，然後返回一個含有3個元素的tuple，(dirpath, dirnames, filenames)，要注意的是，os.walk()會返回一個generater，所以調用的時候一定要放到for迴圈中。
複製代碼 代碼如下:

import os
def walk_dir(dirname):
for root,dirs,files in os.walk(dirname):
for f in files:
yield os.path.join(root, f)

方法二：利用os.listdir()

這個方法也比較簡單，列出目錄然後判斷是不是檔案夾，不是的話直接返迴文件路徑，是的話遞迴調用。
複製代碼 代碼如下:

import os
def walk_dir2(dirname):
for d in os.listdir(dirname):
path = os.path.join(dirname, d)
if os.path.isdir(path):
#note recursive in for...in...
for f in walk_dir2(path):
yield f
else:
yield path

這裡值得一提的是，由於函數中使用了yield，所以會被認為是一個generater，遞迴調用的時候也需要放到for迴圈中，否則函數並不會被真正執行。最新的Python 3.3中引入了yield from文法，可以解決上面的問題。

