|
马上注册,结交更多好友,享用更多功能^_^
您需要 登录 才可以下载或查看,没有账号?立即注册
x
在做30课的课后作业时,自己写的代码在运行的时候会出现MemoryError
还请大佬帮忙看看是否是自己写的代码的问题点。
作业题目是:
用户输入关键字,查找当前文件夹内(如果包含子文件夹,则进入子文件夹继续搜索)所有含有该关键字的文本文件(.txt后缀),要求显示该文件所在位置以及关键字在文件中的具体位置(第几行第几个字符)
我的代码如下:
- import os
- def check_txt(directory, keyword):
- os.chdir(directory)
- all_files = os.listdir()
- for each in all_files:
- if os.path.isfile(os.path.join(directory, each)):
- if os.path.splitext(each)[1] == '.txt':
- f = open(each)
- countline = 0 #统计在第几行
- pos = [] #统计关键字在每行的位置
- for eachline in f:
- countline += 1
- if keyword in eachline:
- countpos = 0
- while countpos < len(eachline):
- pos.append(eachline.find(keyword, countpos))
- countpos = eachline.find(keyword, countpos) + 1
- print('==============================================\n\
- 在文件【%s】中找到关键字【%s】\n\
- 关键字出现在第 %d 行,第 %s 个位置。' % (os.getcwd(), keyword, countline, pos))
- f.close()
- if os.path.isdir(os.path.join(directory, each)):
- check_txt(os.path.join(directory, each), keyword)
- os.chdir(os.pardir)
- directory = input('请输入目标路径:')
- keyword = input('请输入待查找的关键字:')
- decide = input('请问是否需要打印关键字【%s】在文件中的具体位置(YES/NO):' % keyword)
- if decide in ['Yes', 'yes', 'YES']:
- check_txt(directory, keyword)
复制代码
|
|