《入门学习Python》第30讲第4题

shizhou

看了小甲鱼老师的答案，感觉逻辑复杂，看懂有难度。参考后，写了这个程序，也能实现！还望大神指教，是否符合题目要求，哪里还可以优化改进！

1. import os
   
2. 
   
3. def file_key(key_word) :
   
4.      for each_file in os.listdir(os.curdir) :
   
5.         if os.path.splitext(each_file)[1] == '.txt':  
   
6.             if key_word in open(each_file).read():
   
7.                 ext = os.getcwd() + os.sep + each_file
   
8.                 print('在文件【%s】中找到关键字【%s】'%(ext,key_word))
   
9. 
   
10.             f = open(each_file)
    
11.             row = 0
    
12.             for eachline in f:
    
13.                 row += 1
    
14.                 if key_word in eachline:
    
15.                     print('关键字出现在第 %d 行，'% row,end='')
    
16. 
    
17.                     pos = []
    
18.                     begin = eachline.find(key_word)
    
19.                     while begin != -1:
    
20.                         pos.append(begin+1) # 用户的角度是从1开始数
    
21.                         begin = eachline.find(key_word,begin+1) # 从下一个位置继续查找
    
22.                     print('第 %s 个位置。'% pos)
    
23.             print()
    
24.             print(spacing)
    
25.   
    
26.         if os.path.isdir(each_file) :
    
27.             os.chdir(each_file)
    
28.             file_key(key_word)
    
29.             os.chdir(os.pardir) # 递归调用后切记返回上一层目录
    
30. 
    
31. key_word = input('请将该代码放于待查找的文件夹内，请输入关键字：')
    
32. 
    
33. decide = input('请问是否需要打印关键字【%s】在文件中的具体位置（YES/NO）：'% key_word)
    
34. spacing = '====================================================================\n'
    
35. print(spacing)
    
36. 
    
37. if decide in ['YES','Yes','yes']:
    
38.      file_key(key_word)
    
39. 
    

复制代码

ba21

自己都改进了，小甲鱼老师的答案还会感觉逻辑复杂吗？