1200字范文 > python查找单词的位置_Python实现单词查询文件查找

python查找单词的位置_Python实现单词查询文件查找

时间：2020-01-11 14:41:02

最近学C++ Primer，做到第十二章有个习题。要求针对英文文本，对于用户想搜索的单词，打印出该单词在文本中出现的总次数，单词所出现行号及对应的行内容；单词在一行内出现多次，只打印该行一次。C++的代码太长就不给出，实现大概是用vector保存每一行内容，set保存每个单词出现的行号，map来保存单词与所在行号的映射。set.size()表示单词出现的总行数。这是个很好的思路，然而用类的方式实现起来有点繁琐了。好久不用Python了，灵机一动，想试试在Python上如何简单实现同样的功能。

单词查询

程序较为简单，直接上代码。

# 查询文本中某单词出现的次数，并打印其出现的行号及所在行的内容

# 只适用于英文文本

import re

from string import punctuation

text = open('find.txt')

text_list = text.readlines()

# 删除标点及其他常用符号

# punctuation = r"""!"#$%&'()*+,-./:;<=>?@[\]^_`{|}~"""

text_plain = re.sub(r'[{}]'.format(punctuation), '', ''.join(text_list))

# 大写转换为小写，便于正确统计单词数

one_word_list = [word.lower() for word in text_plain.split()]

print(one_word_list)

def run_query(wanted):

word_total = 0

# 统计出现的总个数

for each in one_word_list:

if each == wanted:

word_total += 1

print('"{}" occurs {} times'.format(wanted, word_total))

line_number = 0

for line in text_list:

line_plain = re.sub(r'[{}]'.format(punctuation), '', line)

word_list = [word.lower() for word in line_plain.split()]

# 按照用户习惯第一行从"1"开始

line_number += 1

# 每行的单词列表

if wanted in word_list:

# 而下标"0"表示第一行，故需要减去1

print('\tline {}: {}'.format(line_number, text_list[line_number - 1]), end='')

if __name__ == '__main__':

while True:

sought = input('Input a word you want to search: ')

if sought == 'q':

break

run_query(sought)

看下结果，下面这张是Python下运行，yes出现94次。

C++中，yes出现91次。

仔细观察，两者对应的行号是一样的。为何单词出现次数C++版本就比Python版本少了？原因在于C++ Primer里面的写法是用的set，由于set里面的元素不能重复，如果想要查询的单词在一行内出现多次，实际也只记录一次。程序用set.size()表示单词出现总次数，即假设有n行里存在这个单词，单词出现总次数就是n。不知道作者本来就想实现这样的功能还是出于什么原因，私以为，这不符合我们的初衷。Python版本里对于想查询的单词，不管是否在一行内出现多次，都真正做到了精确计数。

文件查找

突然想起以前还写过个小脚本，查询本地文件。是当时看廖雪峰的Pytthon教程时做的一个课后习题。一起贴在这儿。

import os

def find_files(path, wanted):

try:

# 盘符内所有文件(夹)的路径

dir_list = os.listdir(path)

for filename in dir_list:

# 当前文件(夹)的路径

new_path = os.path.join(path, filename)

# 如果是文件夹，深入下一级继续查找

if os.path.isdir(new_path):