从python中的多个文件中提取cat和grep，并将输出写入一个文件？

import os

# 定义要搜索的关键词
keywords = ['cat', 'grep']

# 定义要读取的文件列表
file_list = ['file1.txt', 'file2.txt', 'file3.txt']

# 定义输出文件名
output_file = 'output.txt'

# 打开输出文件
with open(output_file, 'w') as f_out:
    # 遍历文件列表
    for file_name in file_list:
        # 打开当前文件
        with open(file_name, 'r') as f_in:
            # 逐行读取文件内容
            for line in f_in:
                # 判断当前行是否包含关键词
                if any(keyword in line for keyword in keywords):
                    # 将包含关键词的行写入输出文件
                    f_out.write(line)

# 输出完成提示
print('提取完成，结果已写入文件', output_file)