Python：如何从以单词为键、文件名为值的文件列表构建单词字典？

Question

I have a list of words that I have extracted from several files, and I need to build a dictionary such that the word would be a key corresponding to a list of filenames in which the word appears (can be more than one) as its value.

I have already got code to extract all the words from the files and remove spaces, commas etc. So it is a list of words.

The output should look something like this:

{'on': ['file1.txt'], 'got': ['file1.txt'],'hello': ['file1.txt'],'a': ['file1.txt', 'file2.txt'], 'bad': ['file1.txt', 'file2.txt']}

How can I achieve that?

Answer 1

I guess you want something like this.

Note: This python script file must be located in the same location as the txt files.

files = os.listdir()
txt_files = []
for file in files:
    if file.endswith('.txt'):
        txt_files.append(file)

words = dict()

for file in txt_files:
    with open(file,'r',encoding='utf-8') as file:
        word_list = file.readlines()
        words[str(file.name)] = word_list

print(words)

Answer 2

Hope this works

import os

wordLst = ['on', 'got','hello','a',
       'bad']

dic = {}

for word in wordLst:
    dic.update({word: []})

path = r'yourPath'
filelst = os.listdir(path)

for file in filelst:

if '.txt' not in file: continue

f = open(path + '\\' + file, 'r')
txt = f.readlines()

for key in dic.keys():
    if key in txt:
        dic[key].append(file)