python - 为多个根目录调用 os.walk？

Question

我注意到此处描述的 os.walk API:http://docs.python.org/2/library/os.html是这样的:

for root, dirs, files in os.walk('C:\\Users\\Me\\Videos')

无论如何，我是否会在同一次调用中将多个目录传递给此 API？

Answer 1

你不能只使用两个 for 循环吗？

for x in ("c:\\root1", "c:\\root2", "c:\\root3"):
    for root, dirs, files in os.walk(x):
        pass

或者使用 itertools.chain() 的凌乱单行代码:

for root, dirs, files in itertools.chain(*[os.walk(x) for x in ("c:\\root1", "c:\\root2", "c:\\root3")]):
    pass