30讲第四题 疑问

Isaiah

题目：

                               
登录/注册后可看大图

以下是甲鱼哥的答案：
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)

替换成：
all_file=os.walk(os.curdir)
target_file=[]
for each_file in all_file:
        if os.path.splitext(each_file[2])[1]=='.txt':
            each_file=os.path.join(each_file[0],each_file[2])
            target_file.append(each_file)

报错：AttributeError: 'list' object has no attribute 'rfind'
于是乎我用shell试了下：却是行的！？ 这是什么原因呢？
>>> a
('acv', 'a.txt')
>>> os.path.splitext(a[1])
('a', '.txt')
>>> os.path.splitext(a[1])[1]
'.txt'

===============完整答案==============

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)

新手·ing

我试了一下，没问题啊
楼主改的没错，代码有很多种写法
就是为了简洁

Isaiah

新手·ing 发表于 2017-4-9 12:17
我试了一下，没问题啊
楼主改的没错，代码有很多种写法
就是为了简洁

懂了~
那报错的英文 应该是不用列表做参数的意思
所以each_file[2]是个列表 所有不行

新手·ing

Isaiah 发表于 2017-4-9 12:35
懂了~
那报错的英文 应该是不用列表做参数的意思
所以each_file[2]是个列表 所有不行

所以改吧

新手·ing

Isaiah 发表于 2017-4-9 12:35
懂了~
那报错的英文 应该是不用列表做参数的意思
所以each_file[2]是个列表 所有不行

所以改吧