'file1.txt', 'file2.txt', 'file3.txt' - названия файлов, переданные при создании объекта ['word1', 'word2'], ['word3', 'word4'], ['word5', 'word6', 'word7'] - список слов, найденных в каждом файле соответственно find_words(word) - метод, который принимает на вход слово и возвращает список файлов, в которых это слово встречается. Пример использования: finder = WordsFinder('file1.txt', 'file2.txt', 'file3.txt') finder.get_all_words() finder.find_words('apple')

class WordsFinder: def init(self, *files): self.files = files self.words = [] for file in files: with open(file, 'r') as f: words = f.read().split() self.words.append(words)

def get_all_words(self):
    return self.words

def find_words(self, word):
    result = []
    for i in range(len(self.words)):
        if word in self.words[i]:
            result.append(self.files[i])
    return result

Пример использования

finder = WordsFinder('file1.txt', 'file2.txt', 'file3.txt') finder.get_all_words() finder.find_words('apple')