[已解决]第30讲课后作业

齐大胖

import os

def print_pos(key_dict):
    keys = key_dict.keys()
    keys = sorted(keys) # 由于字典是无序的，我们这里对行数进行排序
    for each_key in keys:
        print('关键字出现在第 %s 行，第 %s 个位置。' % (each_key, str(key_dict[each_key])))


def pos_in_line(line, key):
    pos = []
    begin = line.find(key)
    while begin != -1:
        pos.append(begin + 1) # 用户的角度是从1开始数
        begin = line.find(key, begin+1) # 从下一个位置继续查找

    return pos


def search_in_file(file_name, key):
    f = open(file_name)
    count = 0 # 记录行数
    key_dict = dict() # 字典，用户存放key所在具体行数对应具体位置
   
    for each_line in f:
        count += 1
        if key in each_line:
            pos = pos_in_line(each_line, key) # key在每行对应的位置
            key_dict[count] = pos
   
    f.close()
    return key_dict


def search_files(key, detail):   
    all_files = os.walk(os.getcwd())
    txt_files = []

    for i in all_files:
        for each_file in i[2]:
            if os.path.splitext(each_file)[1] == '.txt': # 根据后缀判断是否文本文件
                each_file = os.path.join(i[0], each_file)
                txt_files.append(each_file)

    for each_txt_file in txt_files:
        key_dict = search_in_file(each_txt_file, key)
        if key_dict:
            print('================================================================')
            print('在文件【%s】中找到关键字【%s】' % (each_txt_file, key))
            if detail in ['YES', 'Yes', 'yes']:
                print_pos(key_dict)


key = input('请将该脚本放于待查找的文件夹内，请输入关键字：')
detail = input('请问是否需要打印关键字【%s】在文件中的具体位置（YES/NO）：' % key)
search_files(key, detail)


这个for each_file in i[2]:  i[2]是什么意思呢 我感觉不加这句好像也行的吧（弱弱问一句）

最佳答案

月排行榜 / 总排行榜

十月故里

2020-2-27 22:10:43

all_files = os.walk(os.getcwd())
这里用的os.walk(top)，遍历top路径以下所有的子目录，返回一个三元组：(路径, [包含目录], [包含文件])
我打印了一个出来
import os
all_files = os.walk(os.getcwd())
list1=list(all_files)
print( list1)
[('F:\\测试', ['文件夹1', '文件夹2'], ['031pickle.py', '1.py', '1.txt', '2.py', '3.py', 'boy_1.txt', 'boy_2.txt', 'boy_3.txt', 'cedioList.txt', 'girl_1.txt', 'girl_2.txt', 'girl_3.txt', 'record.txt', 'vedioList.txt', '新建 Microsoft Office Excel 工作表.xlsx', '新建 Microsoft Office Word 文档.docx']), ('F:\\测试\\文件夹1', ['1'], []), ('F:\\测试\\文件夹1\\1', [], []), ('F:\\测试\\文件夹2', [], ['1.txt'])]
可以看到i[2]指的是文件，不过要注意的是，如果你的脚本所在文件里面也文件夹的话，它会继续进去搜索打印的
for each_file in i[2]:
在执行这一步的时候就可以看到每一个i对应的是里面的一个元组，而元组索引为2的对应的就是文件了

跳转到最佳答案楼层

qiuyouzhi

你可以试试把[2]去掉，看看什么样子

zltzlt

i[2] 是获取文件名。

十月故里

all_files = os.walk(os.getcwd())
这里用的os.walk(top)，遍历top路径以下所有的子目录，返回一个三元组：(路径, [包含目录], [包含文件])
我打印了一个出来
import os
all_files = os.walk(os.getcwd())
list1=list(all_files)
print( list1)
[('F:\\测试', ['文件夹1', '文件夹2'], ['031pickle.py', '1.py', '1.txt', '2.py', '3.py', 'boy_1.txt', 'boy_2.txt', 'boy_3.txt', 'cedioList.txt', 'girl_1.txt', 'girl_2.txt', 'girl_3.txt', 'record.txt', 'vedioList.txt', '新建 Microsoft Office Excel 工作表.xlsx', '新建 Microsoft Office Word 文档.docx']), ('F:\\测试\\文件夹1', ['1'], []), ('F:\\测试\\文件夹1\\1', [], []), ('F:\\测试\\文件夹2', [], ['1.txt'])]
可以看到i[2]指的是文件，不过要注意的是，如果你的脚本所在文件里面也文件夹的话，它会继续进去搜索打印的
for each_file in i[2]:
在执行这一步的时候就可以看到每一个i对应的是里面的一个元组，而元组索引为2的对应的就是文件了