python os.walk到一定级别[重复]

我想构建一个程序，使用一些基本代码来读取文件夹并告诉我该文件夹中有多少文件。 目前我是这样做的：

import os

folders = ['Y:\\path1', 'Y:\\path2', 'Y:\\path3']
for stuff in folders:
    for root, dirs, files in os.walk(stuff, topdown=True):
        print("there are", len(files), "files in", root)

除非“主”文件夹中有多个文件夹，否则这种方法效果很好，因为由于文件夹/文件管理不善，它可能会返回一长串垃圾文件列表。所以我最多只想到第二层。例子：

Main Folder
---file_i_want
---file_i_want
---Sub_Folder
------file_i_want <--*
------file_i want <--*
------Sub_Folder_2
---------file_i_dont_want
---------file_i_dont_want

我知道如何只进入第一级break与del dirs[:]取自这个帖子 https://stackoverflow.com/questions/4117588/non-recursive-os-walk and 还有这个帖子 https://stackoverflow.com/questions/229186/os-walk-without-digging-into-directories-below.

import os
import pandas as pd

folders = ['Y:\\path1', 'Y:\\path2', 'Y:\\path3']
for stuff in folders:
    for root, dirs, files in os.walk(stuff, topdown=True):
        print("there are", len(files), "files in", root)
        del dirs[:] # or a break here. does the same thing.

但无论我如何搜索，我都找不到如何深入两层的方法。我可能只是不理解其他帖子或其他什么？我在想类似的事情del dirs[:2]但无济于事。有人可以指导我或向我解释如何实现这一目标吗？

你可以这样做：

depth = 2

# [1] abspath() already acts as normpath() to remove trailing os.sep
#, and we need ensures trailing os.sep not exists to make slicing accurate. 
# [2] abspath() also make /../ and ////, "." get resolved even though os.walk can returns it literally.
# [3] expanduser() expands ~
# [4] expandvars() expands $HOME
# WARN: Don't use [3] expanduser and [4] expandvars if stuff contains arbitrary string out of your control.
#stuff = os.path.expanduser(os.path.expandvars(stuff)) # if trusted source
stuff = os.path.abspath(stuff)

for root,dirs,files in os.walk(stuff):
    if root[len(stuff):].count(os.sep) < depth:
        for f in files:
            print(os.path.join(root,f))

key is: if root[len(stuff):].count(os.sep) < depth

它删除了stuff from root，所以结果是相对于stuff。只需计算文件分隔符的数量即可。

深度就像findLinux 中的命令，即-maxdepth 0意味着什么都不做，-maxdepth 1仅扫描第一级文件，并且-maxdepth 2扫描文件包含子目录。

当然，它仍然会扫描完整的文件结构，但除非它非常深，否则就可以了。

另一种解决方案是仅使用os.listdir以最大递归级别递归（使用目录检查），但如果您不需要它，那就有点棘手了。由于这并不难，这里有一个实现：

def scanrec(root):
    rval = []

    def do_scan(start_dir,output,depth=0):
        for f in os.listdir(start_dir):
            ff = os.path.join(start_dir,f)
            if os.path.isdir(ff):
                if depth<2:
                    do_scan(ff,output,depth+1)
            else:
                output.append(ff)

    do_scan(root,rval,0)
    return rval

print(scanrec(stuff))  # prints the list of files not below 2 deep

Note: os.listdir and os.path.isfile执行2stat调用所以不是最佳的。在Python 3.5中，使用os.scandir可以避免双重调用。