os.walk無需挖掘到以下目錄

Question

如何限制os.walk只返回我提供的目錄中的文件？

def _dir_list(self, dir_name, whitelist): 
    outputList = [] 
    for root, dirs, files in os.walk(dir_name): 
     for f in files: 
      if os.path.splitext(f)[1] in whitelist: 
       outputList.append(os.path.join(root, f)) 
      else: 
       self._email_to_("ignore") 
    return outputList 

Answer 1

使用walklevel函數。

import os 

def walklevel(some_dir, level=1): 
    some_dir = some_dir.rstrip(os.path.sep) 
    assert os.path.isdir(some_dir) 
    num_sep = some_dir.count(os.path.sep) 
    for root, dirs, files in os.walk(some_dir): 
     yield root, dirs, files 
     num_sep_this = root.count(os.path.sep) 
     if num_sep + level <= num_sep_this: 
      del dirs[:] 

它的工作原理就像os.walk，但你可以傳遞一個level參數，指示遞歸將有多深。

Answer 2

你可以使用os.listdir()返回名稱的列表（文件和目錄）在指定目錄。如果您需要區分文件和目錄，請在每個名稱上撥打os.stat()。

Answer 3

請勿使用os.walk。

實施例：

import os 

root = "C:\\" 
for item in os.listdir(root): 
    if os.path.isfile(os.path.join(root, item)): 
     print item 

Answer 4

使用listdir的建議是一個好。你的問題的直接答案是root, dirs, files = os.walk(dir_name).next()

Answer 5

如果你有比頂級目錄更復雜的要求（例如忽略VCS dirs等），你也可以修改目錄列表來防止os.walk遞歸遍歷它們。

即：

def _dir_list(self, dir_name, whitelist): 
    outputList = [] 
    for root, dirs, files in os.walk(dir_name): 
     dirs[:] = [d for d in dirs if is_good(d)] 
     for f in files: 
      do_stuff() 

注 - 要小心發生變異的名單，而不僅僅是重新綁定。 os.walk顯然不知道外部重新綁定。

Answer 6

你也可以做到以下幾點：

for path, subdirs, files in os.walk(dir_name): 
    for name in files: 
     if path == ".": #this will filter the files in the current directory 
      #code here 

Answer 7

我認爲解決的辦法其實很簡單。

使用

break 

只做for循環第一次迭代，必須有一個更優雅的方式。

for root, dirs, files in os.walk(dir_name): 
    for f in files: 
     ... 
     ... 
    break 
... 

調用os.walk第一次，它返回鬱金香當前目錄​​下，然後在下一循環的下一個目錄的內容。

以原始腳本，只需添加一個休息。

def _dir_list(self, dir_name, whitelist): 
    outputList = [] 
    for root, dirs, files in os.walk(dir_name): 
     for f in files: 
      if os.path.splitext(f)[1] in whitelist: 
       outputList.append(os.path.join(root, f)) 
      else: 
       self._email_to_("ignore") 
     break 
    return outputList 

Answer 8

與listdir同樣的想法，但更短：

[f for f in os.listdir(root_dir) if os.path.isfile(os.path.join(root_dir, f))] 

Answer 9

這是我如何解決它

if recursive: 
    items = os.walk(target_directory) 
else: 
    items = [next(os.walk(target_directory))] 

... 

Answer 10

使用listdir同時，當有一個陷阱。 os.path.isdir（標識符）必須是絕對路徑。要選擇子目錄你這樣做：

for dirname in os.listdir(rootdir): 
    if os.path.isdir(os.path.join(rootdir, dirname)): 
    print("I got a subdirectory: %s" % dirname) 

另一種方法是改變的目錄做的測試，而不os.path.join（）。

Answer 11

在Python 3，我能做到這一點。

import os 
dir = "/path/to/files/" 

#List all files immediately under this folder: 
print (next(os.walk(dir))[2]) 

#List all folders immediately under this folder: 
print (next(os.walk(dir))[1]) 

Answer 12

for path, dirs, files in os.walk('.'): 
    print path, dirs, files 
    del dirs[:] # go only one level deep 

Answer 13

您可以使用此代碼段

for root, dirs, files in os.walk(directory): 
    if level > 0: 
     # do some stuff 
    else: 
     break 
    level-=1 

Answer 14

感覺自己就像扔我的2便士

baselevel = len(rootdir.split("\\")) 
for subdirs, dirs, files in os.walk(rootdir): 
    curlevel = len(subdirs.split("\\")) 
    if curlevel <= baselevel + 1: 
     [do stuff] 

Answer 15

創建排除列表，使用fnmatch跳過目錄結構和d執行過程

excludes= ['a\*\b', 'c\d\e'] 
for root, directories, files in os.walk('Start_Folder'): 
    if not any(fnmatch.fnmatch(nf_root, pattern) for pattern in excludes): 
     for root, directories, files in os.walk(nf_root): 
      .... 
      do the process 
      .... 

相同「包括」：

if **any**(fnmatch.fnmatch(nf_root, pattern) for pattern in **includes**):