如何在大量文件夹中搜索两种类型的文件并比较它们的大小？

import os

def search_and_compare_files(folder_path, file_ext1, file_ext2):
    file_sizes = {}  # 记录文件大小的字典

    # 遍历文件夹
    for root, dirs, files in os.walk(folder_path):
        for file in files:
            file_path = os.path.join(root, file)

            # 判断文件类型
            if file.endswith(file_ext1) or file.endswith(file_ext2):
                # 记录文件大小
                file_size = os.path.getsize(file_path)
                file_sizes[file_path] = file_size

    # 比较文件大小
    if len(file_sizes) >= 2:
        sorted_files = sorted(file_sizes.items(), key=lambda x: x[1])  # 按文件大小排序
        smallest_file = sorted_files[0]
        largest_file = sorted_files[-1]
        print("最小的文件是：", smallest_file[0], "，大小为：", smallest_file[1], "字节")
        print("最大的文件是：", largest_file[0], "，大小为：", largest_file[1], "字节")
    else:
        print("未找到足够数量的文件")

# 示例用法
folder_path = "/path/to/folder"
file_ext1 = ".txt"
file_ext2 = ".doc"
search_and_compare_files(folder_path, file_ext1, file_ext2)