[已解决]python代码求优化

sunyt

代码目的是找出对应类型的文件，可结果里会有不相干的文件名，甚至有名字不相干的目录名

1. import os
   
2. 
   
3. def file_find(dir_position, suffix):
   
4.     os.chdir(dir_position)
   
5.     all_files = os.listdir(os.curdir)
   
6. 
   
7.     for each_file in all_files:
   
8.         if os.path.splitext(each_file)[1] in suffix:
   
9.             viewlist.append(os.getcwd() + os.sep + each_file + '\n')
   
10.         if os.path.isdir(each_file):
    
11.             file_find(each_file, suffix)
    
12.             os.chdir(os.pardir)
    
13. 
    
14. dir_position = input('请输入路径:')
    
15. suffix = input('请输入要查找的文件类型(.txt):')
    
16. 
    
17. viewlist = []
    
18. 
    
19. file_find(dir_position, suffix)
    
20. 
    
21. f = open(dir_position + os.sep + 'viewlist.txt', 'w')
    
22. f.writelines(viewlist)
    
23. f.close()
    

复制代码

圈出来的是目录，而且名字都没有关系，为什么会打印出来
file:///C:/Users/123/Desktop/Inked%E6%8D%95%E8%8E%B7_LI.jpg

最佳答案

月排行榜 / 总排行榜

大马强

2021-9-14 12:12:28

1. import os
   
2. 
   
3. 
   
4. def file_find(dir_position, suffix):
   
5.     os.chdir(dir_position)
   
6.     all_files = os.listdir(os.curdir)
   
7. 
   
8.     for each_file in all_files:
   
9.         if os.path.isdir(each_file):
   
10.             file_find(each_file, suffix)
    
11.             os.chdir(os.pardir)
    
12.         else:
    
13.             if os.path.splitext(each_file)[1] in suffix:
    
14.                 viewlist.append(os.getcwd() + os.sep + each_file + '\n')
    
15. 
    
16. 
    
17. dir_position = input('请输入路径:')
    
18. suffix = input('请输入要查找的文件类型(.txt):')
    
19. 
    
20. viewlist = []
    
21. 
    
22. file_find(dir_position, suffix)
    
23. 
    
24. f = open(dir_position + os.sep + 'viewlist.txt', 'w', encoding="utf-8")
    
25. f.writelines(viewlist)
    
26. f.close()
    

复制代码

跳转到最佳答案楼层

sunyt

怎么发图片啊

大马强

新人发图三件套 官方教程 图壳 图桌

blahblahfc

可能是第8行:

1.         if os.path.splitext(each_file)[1] in suffix:

复制代码

无后缀文件或文件夹时 os.path.splitext(each_file)[1] 返回空字符串 ''，
导致 if 条件 '' in suffix 一直为真，suffix 的值是字符串

1. os.path.splitext('foo')
   
2. # 返回 ('foo', '')

复制代码

大马强

1. import os
   
2. 
   
3. 
   
4. def file_find(dir_position, suffix):
   
5.     os.chdir(dir_position)
   
6.     all_files = os.listdir(os.curdir)
   
7. 
   
8.     for each_file in all_files:
   
9.         if os.path.isdir(each_file):
   
10.             file_find(each_file, suffix)
    
11.             os.chdir(os.pardir)
    
12.         else:
    
13.             if os.path.splitext(each_file)[1] in suffix:
    
14.                 viewlist.append(os.getcwd() + os.sep + each_file + '\n')
    
15. 
    
16. 
    
17. dir_position = input('请输入路径:')
    
18. suffix = input('请输入要查找的文件类型(.txt):')
    
19. 
    
20. viewlist = []
    
21. 
    
22. file_find(dir_position, suffix)
    
23. 
    
24. f = open(dir_position + os.sep + 'viewlist.txt', 'w', encoding="utf-8")
    
25. f.writelines(viewlist)
    
26. f.close()
    

复制代码

大马强

你那样的逻辑会一些问题，我也很难说清楚
一般我用的逻辑是

1. if 是否为文件夹:
   
2.     xxx
   
3. else:
   
4.     if 是否满足要求:
   
5.         xxxx

复制代码

大马强

大马强 发表于 2021-9-14 12:18
你那样的逻辑会一些问题，我也很难说清楚
一般我用的逻辑是

因为不是文件夹，就是一个文件，所以用if else 搭配就不用两次判断

sunyt

大马强 发表于 2021-9-14 12:04
新人发图三件套 官方教程 图壳 图桌

感谢