python - 我们如何打印 Python 文本文件中出现单词的行号？

Question

我需要它来打印文本文件中相应的行号。

def index (filename, lst):
    infile = open('raven.txt', 'r')
    lines =  infile.readlines()
    words = []
    dic = {}

    for line in lines:
        line_words = line.split(' ')
        words.append(line_words)
    for i in range(len(words)):
        for j in range(len(words[i])):
            if words[i][j] in lst:

                dic[words[i][j]] = i

    return dic

结果:

In: index('raven.txt',['raven', 'mortal', 'dying', 'ghost', 'ghastly', 'evil', 'demon'])

Out: {'dying': 8, 'mortal': 29, 'raven': 77, 'ghost': 8}

(上面的文字出现在几行中，但它只打印一行，有些则不打印任何内容 此外，它不计算文本文件中的空行。所以 8 实际上应该是 9，因为有一个空行不计算在内。)

请告诉我如何解决这个问题。

Answer 1

def index (filename, lst):

    infile = open('raven.txt', 'r')
    lines =  infile.readlines()
    words = []
    dic = {}

    for line in lines:
        line_words = line.split(' ')
        words.append(line_words)
    for i in range(len(words)):
        for j in range(len(words[i])):
            if words[i][j] in lst:
                if words[i][j] not in dic.keys():
                    dic[words[i][j]] = set()
                dic[words[i][j]].add(i + 1) #range starts from 0
    return dic

如果单词在同一行中多次出现，则使用集合而不是列表很有用。