[已解决]python查找.txt文件中想要的关键字的准确位置

小甲鱼的铁粉

编写一个程序，用户输入关键字，查找当前文件夹内（如果当前文件夹内包含文件夹，则进入文件夹继续搜索）所有含有该关键字的文本文件.txt后缀，要求显示该文件所在的位置以及关键字在文件中的具体位置第几行第几个字符

我写到一半了，在.txt文件中找到关键字的准确位置这里卡住了，我能找的关键字在第几行，但是怎么找他在第几行的第几个位置呢

这是我写了一半的代码

1. import os
   
2. def search_text(start_dir,aim):
   
3. 
   
4.     os.chdir(start_dir)
   
5. 
   
6.     for each_file in os.listdir(os.curdir):
   
7. 
   
8.         if (behind = os.path.splitext(each_file)[1]) == '.txt':#判断后缀名是否为.txt
   
9. 
   
10.             search_aim(each_file)#调用在.txt文件中找准确位置的函数，但是我没写，不知道这个函数怎么写
    
11.             
    
12.         if os.path.isdir(erch_file):#判断是否为文件夹，是的话进文件夹查找
    
13.             
    
14.             search_text(each_file,aim)
    
15. 
    
16.             os.chdir(os.pardir)#返回上级目录
    
17. 
    
18. 
    
19. def search_aim(each_file):
    
20. 
    
21.    
    
22.    
    
23. 
    
24. aim = input("请输入要查找的内容")
    
25. start_dir = input("情输入开始查找的路径")
    

复制代码

这个是答案，看着好麻烦 ，好多看不懂的

1. 
   
2. import os
   
3. 
   
4. 
   
5. 
   
6. def print_pos(key_dict):
   
7. 
   
8.     keys = key_dict.keys()
   
9. 
   
10.     keys = sorted(keys) # 由于字典是无序的，我们这里对行数进行排序
    
11. 
    
12.     for each_key in keys:
    
13. 
    
14.         print('关键字出现在第 %s 行，第 %s 个位置。' % (each_key, str(key_dict[each_key])))
    
15. 
    
16. 
    
17. 
    
18. 
    
19. 
    
20. def pos_in_line(line, key):
    
21. 
    
22.     pos = []
    
23. 
    
24.     begin = line.find(key)
    
25. 
    
26.     while begin != -1:
    
27. 
    
28.         pos.append(begin + 1) # 用户的角度是从1开始数
    
29. 
    
30.         begin = line.find(key, begin+1) # 从下一个位置继续查找
    
31. 
    
32. 
    
33. 
    
34.     return pos
    
35. 
    
36. 
    
37. 
    
38. 
    
39. 
    
40. def search_in_file(file_name, key):
    
41. 
    
42.     f = open(file_name)
    
43. 
    
44.     count = 0 # 记录行数
    
45. 
    
46.     key_dict = dict() # 字典，用户存放key所在具体行数对应具体位置
    
47. 
    
48. 
    
49. 
    
50.     for each_line in f:
    
51. 
    
52.         count += 1
    
53. 
    
54.         if key in each_line:
    
55. 
    
56.             pos = pos_in_line(each_line, key) # key在每行对应的位置
    
57. 
    
58.             key_dict[count] = pos
    
59. 
    
60. 
    
61. 
    
62.     f.close()
    
63. 
    
64.     return key_dict
    
65. 
    
66. 
    
67. 
    
68. 
    
69. 
    
70. def search_files(key, detail):   
    
71. 
    
72.     all_files = os.walk(os.getcwd())
    
73. 
    
74.     txt_files = []
    
75. 
    
76. 
    
77. 
    
78.     for i in all_files:
    
79. 
    
80.         for each_file in i[2]:
    
81. 
    
82.             if os.path.splitext(each_file)[1] == '.txt': # 根据后缀判断是否文本文件
    
83. 
    
84.                 each_file = os.path.join(i[0], each_file)
    
85. 
    
86.                 txt_files.append(each_file)
    
87. 
    
88. 
    
89. 
    
90.     for each_txt_file in txt_files:
    
91. 
    
92.         key_dict = search_in_file(each_txt_file, key)
    
93. 
    
94.         if key_dict:
    
95. 
    
96.             print('================================================================')
    
97. 
    
98.             print('在文件【%s】中找到关键字【%s】' % (each_txt_file, key))
    
99. 
    
100.             if detail in ['YES', 'Yes', 'yes']:
     
101. 
     
102.                 print_pos(key_dict)
     
103. 
     
104. 
     
105. 
     
106. 
     
107. 
     
108. key = input('请将该脚本放于待查找的文件夹内，请输入关键字：')
     
109. 
     
110. detail = input('请问是否需要打印关键字【%s】在文件中的具体位置（YES/NO）：' % key)
     
111. 
     
112. search_files(key, detail)
     

复制代码

接下来这个是期望的运行结果

最佳答案

月排行榜 / 总排行榜

Twilight6

2020-6-19 11:53:25

小甲鱼的铁粉 发表于 2020-6-19 11:51
嗯，改了一下
现在的代码是

文件夹里面的 txt 文件编码重新设置下就好了：

第一步：

                               
登录/注册后可看大图

第二步：

                               
登录/注册后可看大图

跳转到最佳答案楼层

小甲鱼的铁粉

这个帖子太长了，麻烦鱼油们看一下吧，辛苦了

yhhpf

1. def search_in_file(file_name, key):
   
2. 
   
3.     f = open(file_name)
   
4. 
   
5.     count = 0 # 记录行数
   
6. 
   
7.     key_dict = dict() # 字典，用户存放key所在具体行数对应具体位置
   
8. 
   
9. 
   
10. 
    
11.     for each_line in f:
    
12. 
    
13.         count += 1
    
14. 
    
15.         if key in each_line:
    
16. 
    
17.             pos = pos_in_line(each_line, key) # key在每行对应的位置
    
18. 
    
19.             key_dict[count] = pos

复制代码

答案里这一段不是写的很明确了么？找到文件后，用count记录行数，逐行查找并记录位置，找到后以字典的形式存储{[count:pos]}

yhhpf

1. def search_in_file(file_name, key):
   
2. 
   
3.     f = open(file_name)
   
4. 
   
5.     count = 0 # 记录行数
   
6. 
   
7.     key_dict = dict() # 字典，用户存放key所在具体行数对应具体位置
   
8. 
   
9. 
   
10. 
    
11.     for each_line in f:#可以理解为：将文件看做一个由多行组成的列表，让each_line代表其中的行，循环每一行
    
12. 
    
13.         count += 1#每次循环count增加1，用以记录第几行
    
14. 
    
15.         if key in each_line:#判断kye是否在该行内
    
16. 
    
17.             pos = pos_in_line(each_line, key) # key在每行对应的位置 #如果在了，则将key在该行所在的位置赋值给pos
    
18. 
    
19.                   key_dict[count] = pos

复制代码

#字典记录对应行数以及位置

Twilight6

你的代码错了俩个地方：

if (behind = os.path.splitext(each_file)[1]) == '.txt':

还有这打错变量名了，each_file：

if os.path.isdir(erch_file):

小甲鱼的铁粉

Twilight6 发表于 2020-6-19 11:45
你的代码错了俩个地方：

嗯，改了一下
现在的代码是

1. import os
   
2. def search_file(start_dir,aim):
   
3. 
   
4.     os.chdir(start_dir)
   
5. 
   
6.     for each_file in os.listdir(os.curdir):
   
7. 
   
8.         behind = os.path.splitext(each_file)[1]
   
9. 
   
10.         if behind  == '.txt':#判断后缀名是否为.txt
    
11.             
    
12.             search_text(each_file,aim)#调用在.txt文件中找准确位置的函数，但是我没写，不知道这个函数怎么写
    
13.             
    
14.         if os.path.isdir(each_file):#判断是否为文件夹，是的话进文件夹查找
    
15. 
    
16.    
    
17.             search_file(each_file,aim)
    
18. 
    
19.             os.chdir(os.pardir)#返回上级目录
    
20. 
    
21. 
    
22. def search_text(each_file,aim):
    
23.     count = 1
    
24.     print(each_file)
    
25.     f1 = open(each_file,encoding = 'utf-8')
    
26.    
    
27.     for each_line in f1:
    
28.         if aim in each_line:
    
29.            print("在%s文件中的%d行找到了%s" % (each_file,count,aim))
    
30.         count += 1
    
31. 
    
32.     f1.close()
    
33.    
    
34. 
    
35. aim = input("请输入要查找的内容")
    
36. start_dir = input("情输入开始查找的路径")
    
37. search_file(start_dir,aim)
    

复制代码

报错了，好像是格式不对
Traceback (most recent call last):
  File "C:\Users\86188\Desktop\1111.py", line 37, in <module>
    search_file(start_dir,aim)
  File "C:\Users\86188\Desktop\1111.py", line 17, in search_file
    search_file(each_file,aim)
  File "C:\Users\86188\Desktop\1111.py", line 12, in search_file
    search_text(each_file,aim)#调用在.txt文件中找准确位置的函数，但是我没写，不知道这个函数怎么写
  File "C:\Users\86188\Desktop\1111.py", line 27, in search_text
    for each_line in f1:
  File "C:\Users\86188\AppData\Local\Programs\Python\Python38-32\lib\codecs.py", line 322, in decode
    (result, consumed) = self._buffer_decode(data, self.errors, final)
UnicodeDecodeError: 'utf-8' codec can't decode byte 0xb2 in position 153: invalid start byte

Twilight6

小甲鱼的铁粉 发表于 2020-6-19 11:51
嗯，改了一下
现在的代码是

文件夹里面的 txt 文件编码重新设置下就好了：

第一步：

                               
登录/注册后可看大图

第二步：

                               
登录/注册后可看大图