python遍历文件遇到的问题

netdzb

def getFiles(dir, suffix): # 查找根目录，文件后缀
    res = []
    for root, directory, files in os.walk(dir):  # =>当前根,根下目录,目录下的文件
            for filename in files:
                    name, suf = os.path.splitext(filename) # =>文件名,文件后缀
                    if suf == suffix:
                            res.append(os.path.join(root, filename)) # =>吧一串字符串组合成路径
    return res


for file_path in getFiles("./", '.png'):  # =>查找以.png结尾的文件
#        img1=Image.open(file_path)
        print(file_path)

当前路径有很多png文件，同时有个子目录101-1，里面也有很多png文件。
我的这段代码会把当前目录下的png和子文件夹的101-1全部列出来的。
如何修改代码，我只要列出当前路径下的png文件，谢谢！

ivor

os.listdir(path='.') 替换 os.walk()

https://docs.python.org/3.8/library/os.html#os.listdir

Gin_Q

1. directory.clear()

复制代码

Gin_Q

我正好也需要，拿去自己改改！

1. #!/usr/bin/env python3
   
2. #coding=utf-8
   
3. 
   
4. import os
   
5. import time
   
6. import shutil
   
7. 
   
8. 
   
9. #禁止访问目录列表
   
10. stop = ['837761056','1289832516','1551961057']
    
11. #指定复制文件列表
    
12. filetype = ['.mp4','.avi']
    
13. #复制或移动目标目录
    
14. dstdirs = r'd:\\'
    
15. 
    
16. for root,dirname,filename in os.walk(os.getcwd()):
    
17.     print('当前目录为：',root)
    
18.     #print('当前目录文件夹有：',dirname)
    
19.     #print('当前文件有：',filename)
    
20.     #获取当前目录文件不递归
    
21.     #dirname.clear()
    
22.     #排除指定目录
    
23.     for i in stop:  
    
24.         if i in dirname:
    
25.             dirname.remove(i)
    
26.    
    
27.     for i in (os.path.join(root,name) for name in filename):
    
28.         #print('完整文件路径: ',i,(os.path.getctime(i)))
    
29.         for ft in filetype:
    
30.             if ft in i:
    
31.                 print('正在复制:\n',i,'--->>>',dstdirs)
    
32.                 #复制
    
33.                 shutil.copy(i,dstdirs)
    
34.                 #print('正在移动:\n',i,'--->>>',dstdirs)
    
35.                 #移动
    
36.                 #shutil.move(i,dstdirs)
    
37.    
    
38. print('Done!')
    
39. os.popen('pause')
    

复制代码