如何从r目录中删除特定扩展名文件

import os

def delete_files_with_extension(directory, extension):
    for root, dirs, files in os.walk(directory):
        for file in files:
            if file.endswith(extension):
                file_path = os.path.join(root, file)
                os.remove(file_path)

        for dir in dirs:
            delete_files_with_extension(os.path.join(root, dir), extension)

# 调用函数删除r目录中的.txt文件
delete_files_with_extension('r', '.txt')