如何编写遍历文件夹并使用给定参数创建新文件/文件内容的脚本

import os

def create_files(folder_path, file_extension, content):
    for root, dirs, files in os.walk(folder_path):
        for file in files:
            if file.endswith(file_extension):
                file_path = os.path.join(root, file)
                with open(file_path, 'w') as f:
                    f.write(content)

# 示例调用
folder_path = '/path/to/folder'  # 需要遍历的文件夹路径
file_extension = '.txt'  # 文件扩展名
content = 'This is the content of the new file.'  # 新文件的内容

create_files(folder_path, file_extension, content)