[已解决]遍历文件夹下的子文件夹，不继续下一级

jackz007

1. import os
   
2. 
   
3. def search(path , file , c):
   
4.     if c < 2:
   
5.         try:
   
6.             for each in os . listdir(path):
   
7.                 x =  os . path . join(path , each)
   
8.                 if os . path . isfile(x):
   
9.                     if each . lower() == file . lower():
   
10.                         print(x)
    
11.                 elif os . path . isdir(x):
    
12.                     search(x , file , c + 1)
    
13.         except Exception as e:
    
14.             print(e)
    
15. path = input('输入初始搜索路径 : ') . strip()
    
16. if path:
    
17.     file = input('输入文件名 : ') . strip()
    
18.     if file:
    
19.         search(path , file , 0)

复制代码

        运行实况：

1. D:\[00.Exercise]\[Python]>python x.py
   
2. 输入初始搜索路径 : C:\
   
3. 输入文件名 : notepad.exe
   
4. [WinError 5] 拒绝访问。: 'C:\\Documents and Settings'
   
5. [WinError 5] 拒绝访问。: 'C:\\Recovery'
   
6. [WinError 5] 拒绝访问。: 'C:\\System Volume Information'
   
7. C:\Windows\notepad.exe
   
8. 
   
9. D:\[00.Exercise]\[Python]>

复制代码

南归

本主题需向作者支付 3 鱼币 才能浏览

编程小白艾雪儿

南归 发表于 2022-6-13 14:52
本主题需向作者支付 3 鱼币 才能浏览

发错了555，应该悬赏的

编程小白艾雪儿

jackz007 发表于 2022-6-13 14:50
运行实况：

输入初始搜索路径是输入大的文件名，文件名是子文件吗
菜鸟表示还不太清楚qvq

Lai013

“想要获取fold文件夹的子文件夹，但不想获取子文件夹内的文件”
如果是想要文件夹名称可以用next()方法

1. print(next(os.walk(fold))[1])

复制代码

如果是想要完整的文件夹路径只循环一次一可以了

1. for root, dirs, files in os.walk(fold):
   
2.     for d in dirs:
   
3.         print(os.path.join(root,d))
   
4.     break

复制代码

“目的是想对大文件夹里的每个小文件夹进行循环，把每个子文件夹中的图片插入word模板里”

1. for root, dirs, files in os.walk(fold):
   
2.     for file in files:
   
3.         if file.split(".")[-1] in ["png","jpg"]:#加入图片格式
   
4.             print(os.path.join(root,file))

复制代码

编程小白艾雪儿

Lai013 发表于 2022-6-13 15:54
“想要获取fold文件夹的子文件夹，但不想获取子文件夹内的文件”
如果是想要文件夹名称可以用next()方法
...

1. #遍历子文件夹
   
2. for root, dirs, files in os.walk(fold):
   
3.     for i in range(0,len(dirs)):        
   
4.         
   
5.         #遍历子文件夹中的图片，每个子文件夹建立一个word
   
6.         for root, dirs, pics in os.walk(dirs[i]):            
   
7.             for j in range(0,len(pics)):

复制代码

想请教下可以这样建立循环吗，    for i in range(0,len(dirs)):        这一行的len(dirs）好像不太对

解决啦！

1. for root, dirs, files in os.walk(fold):
   
2.     for d in dirs:
   
3.         doc=Document()
   
4.         doc_name = os.path.basename(d)
   
5.         for root, dirs, pics in os.walk(d):
   
6.             for i in range(0,len(pics)):

复制代码

Lai013

编程小白艾雪儿 发表于 2022-6-13 17:12
想请教下可以这样建立循环吗，    for i in range(0,len(dirs)):        这一行的len(dirs）好像不太 ...

可以
但是不能用dirs[i], dir[i]返回的是文件夹名字，不包含路径，应该改成os.path.join(root,dirs[i])

还有你这样循环套循环的不好，如果文件夹多的话你的套多少个for循环呢
os.walk()函数就可以遍历文件夹下的所有文件和文件夹了呀，不用for os.walk()后再套for  os.walk()

编程小白艾雪儿

Lai013 发表于 2022-6-13 17:30
可以
但是不能用dirs, dir返回的是文件夹名字，不包含路径，应该改成os.path.join(root,dirs)

如果读取子文件夹不再套循环的话，那就是得加条件来做吧，这是完整的代码，想求问可以怎样优化555，小白能做出来已经极限了

1. #fold中含有子文件夹，子文件夹中含有图片，需要把各个子文件夹中的图片分别插入word
   
2. from docx import Document
   
3. from docx.shared import Inches
   
4. import os
   
5. from PIL import Image
   
6. 
   
7. 
   
8. # 要插入的图片所在的文件夹
   
9. fold=r"C:\\Users\liqian6\Desktop\python\digao\诚信核查"
   
10. 
    
11. #遍历fold中的子文件夹，提取子文件夹名称作为新建word的名称(doc_name)
    
12. for root, dirs, files in os.walk(fold):
    
13.     for d in dirs:
    
14.         doc=Document()
    
15.         doc_name = os.path.basename(d)
    
16.         
    
17.         #遍历子文件夹中的图片，每个子文件夹建立一个word
    
18.         for root, dirs, pics in os.walk(d):
    
19.             for i in range(0,len(pics)):
    
20.                 picpath = d+'\\'+pics[i]
    
21.                 pic_name = os.path.basename(picpath)  
    
22.                
    
23.                 #跳过子文件夹中有PDF文件
    
24.                 if (picpath.endswith('.pdf')):
    
25.                     continue                     
    
26.                
    
27.                 try:
    
28.                     #将图片的名字也插入word中
    
29.                     doc.add_paragraph(pic_name)
    
30.                     #插入图片
    
31.                     doc.add_picture(picpath,width=Inches(6),height=Inches(4))
    
32.                 except Exception:
    
33.                     pic_tmp=Image.open(picpath)
    
34.                     # 如果格式有问题，就用save转换成默认的jpg格式
    
35.                     pic_tmp.save(pic_tmp)
    
36.                     # 把处理后的图片放进Document变量doc中
    
37.                     doc.add_picture(picpath, width=Inches(6),height=Inches(4))
    
38.                 doc.save(doc_name+'.docx')
    
39.       
    
40.             #输出保存成功的标志
    
41.             print(doc_name+"successfully added.")
    
42.         

复制代码

Lai013

编程小白艾雪儿 发表于 2022-6-13 17:49
如果读取子文件夹不再套循环的话，那就是得加条件来做吧，这是完整的代码，想求问可以怎样优化555，小白 ...

能看出你对os.walk()的返回内容不是太了解
os.walk()返回的是一个三元组的生成器
第一个root是目录
第二个dirs是文件夹名称（不包含目录）
第三个files是文件名称（不包含目录）
所以
doc_name = os.path.basename(d) 完全没有必要，doc_name 和d 是一样的
picpath = d+'\\'+pics[i] 改成 picpath = os.path.join(root,pics)
程序能正确的跑起来就可以了
至于优化我觉你自己多想想，毕竟自己相处来的收获最大。

编程小白艾雪儿

Lai013 发表于 2022-6-13 19:41
能看出你对os.walk()的返回内容不是太了解
os.walk()返回的是一个三元组的生成器
第一个root是目录

嗯嗯好嘞！解释的很清楚，非常感谢啦，因为突然有需求中途学习的，确实还不深入。道阻且长，感谢道友哈哈哈