如何根据文件夹的大小以KB、MB或GB为单位显示文件夹的大小？

import os

def get_folder_size(folder_path):
    total_size = 0
    for path, dirs, files in os.walk(folder_path):
        for f in files:
            fp = os.path.join(path, f)
            total_size += os.path.getsize(fp)
    return total_size

def format_size(size):
    # 单位换算
    kilobyte = 1024
    megabyte = kilobyte * 1024
    gigabyte = megabyte * 1024

    if size < kilobyte:
        return f"{size} Bytes"
    elif size < megabyte:
        return f"{size / kilobyte:.2f} KB"
    elif size < gigabyte:
        return f"{size / megabyte:.2f} MB"
    else:
        return f"{size / gigabyte:.2f} GB"

folder_path = "路径/到/文件夹"
folder_size = get_folder_size(folder_path)
formatted_size = format_size(folder_size)

print(f"文件夹大小: {formatted_size}")