关于课后作业030讲第4题

丿autistic

我写的代码提示MemoryError，但是用甲鱼的答案却不会有这种情况，求助问题出在了哪里？怎么解决

1. import os
   
2. 
   
3. 
   
4. def search(key):
   
5.     files = os.walk(os.getcwd())  # 列出路径中的每一个文件
   
6.     txt = []  # 空的列表用于放txt文件
   
7.     for each_file in files:
   
8.         for each in each_file[2]:  # 文件后缀判断是不是txt
   
9.             if os.path.splitext(each)[1] == '.txt':
   
10.                 txt.append(os.path.join(each_file[0], each))
    
11. 
    
12.     for each_txt in txt:  # 对于每一个txt文件
    
13.         f = open(each_txt)
    
14.         f_all = f.read()
    
15.         if key in f_all:
    
16.             f.seek(0, 0)
    
17.             f_search = list(f.readlines())  # 以列表形式列出每一行
    
18.             f.close()
    
19.             local = dict()  # 空的字典用于放位置
    
20.             for i in range(len(f_search)):
    
21.                 if key in f_search[i]:  # 对于列表中的每一个元素查找关键字
    
22.                     pos = []
    
23.                     local.setdefault(i + 1, 0)
    
24.                     begin = f_search[i].find(key)  # 查找关键字第一次出现的位置
    
25.                     while begin != -1:
    
26.                         pos.append(begin + 1)
    
27.                         begin = f_search[i].find(key, begin)  # 查找后续出现的位置
    
28.                     local[i + 1] = pos
    
29.                     if choose.upper() == 'YES':
    
30.                         print('==========')
    
31.                         print('在文件【%s】中找到关键字【%s】' % (each_txt, key))
    
32.                         for each_key in local.keys():
    
33.                             print('关键字出现在第【%s】行，第【%s】个位置' % (each_key, str(local[each_key])))
    
34. 
    
35. key = input('请将该脚本放入待查找的文件夹内，请输入关键字:')
    
36. choose = input('请问是否需要打印关键字【%s】在文件中出现的具体位置(YES/NO):' % key)
    
37. search(key)
    

复制代码

徐颖科

 运行没错误。给一下具体的截图？

丿autistic

徐颖科 发表于 2020-6-18 22:46
 运行没错误。给一下具体的截图？

[img][/img]
显示这样的，是我的电脑问题吗？你运行我的程序能输出正确结果吗？

丿autistic

徐颖科 发表于 2020-6-18 22:46
 运行没错误。给一下具体的截图？

就是这样