Рассортировать файл с ключами 2

# -*- coding: UTF-8 -*-

# Путь к txt файлу ключей
# кодировка UTF-8, по одному в строке
key_file_path = r'E:\fun\key.txt'
# Какие ключи искать
keys = ('хуй', 'хуи', 'пизда')
# Открываем файл с ключами и читаем все строки
key_file = open(key_file_path, encoding='utf-8'
    ).readlines()
# Выполняем для каждого заданного ключа
for key in keys:
    # Создаем файл с именем "ключ.txt"
    new_key_file = open(key+'.txt', 'w', encoding='utf-8')
    # Поиск всех вхождений построчно и запись в файл
    for string_from_file in key_file:
        if key in string_from_file:
            new_key_file.write(string_from_file)