在目录和子目录中查找带扩展名的文件？

Question

我想制作一个Python脚本，以便将我桌面上的文件快速组织到基于扩展名的文件夹中。基本上，我如何使用循环来获取文件，对它做些什么，转到下一个文件，等等？你需要

Answer 1

一切都可能包含在os库，更具体的os.path位的它和shutil之一。

要探索目录树，您可以使用os.walk并移动文件可以使用shutil.move。

---

编辑：我一起砍死一个小脚本让你去：

import os 
import shutil as sh 
from collections import defaultdict 

DESKTOP = '/home/mac/Desktop' 

#This dictionary will contain: <extension>: <list_of_files> mappings 
register = defaultdict(list) 

#Populate the register 
for dir_, dirs, fnames in os.walk('/home/mac/Desktop'): 
    for fname in fnames: 
     register[fname.rsplit('.', 1)[1]].append(os.path.join(dir_, fname)) 

#Iterate over the register, creating the directory and moving the files 
#with that extension in it. 
for dirname, files in register.iteritems(): 
    dirname = os.path.join(DESKTOP, dirname) 
    if not os.path.exists(dirname): 
     os.makedirs(dirname) 
    for file_ in files: 
     sh.move(file_, dirname) 

Answer 2

我会从模块os和文件名的list