30讲课后习题4

雨留下的印记

因为写不出来，就直接看答案了。下面是小甲鱼给的代码

1. import os
   
2. 
   
3. def print_pos(key_dict):
   
4.     keys = key_dict.keys()
   
5.     keys = sorted(keys) # 由于字典是无序的，我们这里对行数进行排序
   
6.     for each_key in keys:
   
7.         print('关键字出现在第 %s 行，第 %s 个位置。' % (each_key, str(key_dict[each_key])))
   
8. 
   
9. 
   
10. def pos_in_line(line, key):
    
11.     pos = []
    
12.     begin = line.find(key)
    
13.     while begin != -1:
    
14.         pos.append(begin + 1) # 用户的角度是从1开始数
    
15.         begin = line.find(key, begin+1) # 从下一个位置继续查找
    
16. 
    
17.     return pos
    
18. 
    
19. 
    
20. def search_in_file(file_name, key):
    
21.     f = open(file_name)
    
22.     count = 0 # 记录行数
    
23.     key_dict = dict() # 字典，用户存放key所在具体行数对应具体位置
    
24.    
    
25.     for each_line in f:
    
26.         count += 1
    
27.         if key in each_line:
    
28.             pos = pos_in_line(each_line, key) # key在每行对应的位置
    
29.             key_dict[count] = pos
    
30.    
    
31.     f.close()
    
32.     return key_dict
    
33. 
    
34. 
    
35. def search_files(key, detail):   
    
36.     all_files = os.walk(os.getcwd())
    
37.     txt_files = []
    
38. 
    
39.     for i in all_files:
    
40.         for each_file in i[2]:
    
41.             if os.path.splitext(each_file)[1] == '.txt': # 根据后缀判断是否文本文件
    
42.                 each_file = os.path.join(i[0], each_file)
    
43.                 txt_files.append(each_file)
    
44. 
    
45.     for each_txt_file in txt_files:
    
46.         key_dict = search_in_file(each_txt_file, key)
    
47.         if key_dict:
    
48.             print('================================================================')
    
49.             print('在文件【%s】中找到关键字【%s】' % (each_txt_file, key))
    
50.             if detail in ['YES', 'Yes', 'yes']:
    
51.                 print_pos(key_dict)
    
52. 
    
53. 
    
54. key = input('请将该脚本放于待查找的文件夹内，请输入关键字：')
    
55. detail = input('请问是否需要打印关键字【%s】在文件中的具体位置（YES/NO）：' % key)
    
56. search_files(key, detail)
    

复制代码

以上是小甲鱼给的代码，但我运行的时候回报错。


请将该脚本放于待查找的文件夹内，请输入关键字：小甲鱼
请问是否需要打印关键字【小甲鱼】在文件中的具体位置（YES/NO）：YES
Traceback (most recent call last):
  File "C:\Users\cwj\Desktop\python练习\seek3.py", line 56, in <module>
    search_files(key, detail)
  File "C:\Users\cwj\Desktop\python练习\seek3.py", line 46, in search_files
    key_dict = search_in_file(each_txt_file, key)
  File "C:\Users\cwj\Desktop\python练习\seek3.py", line 25, in search_in_file
    for each_line in f:
UnicodeDecodeError: 'gbk' codec can't decode byte 0xae in position 4: illegal multibyte sequence

以上为报错结果，以前出现‘gbk’这种错位的时候，一般加上encoding=‘UTF-8’ 就能解决，这次确不行
求大佬告知下该怎么改代码。
我用的python是3.8.2

zltzlt

https://fishc.com.cn/thread-158932-1-1.html

qiuyouzhi

可以使用Notepad++修改编码。
具体方法百度