Python 计算文件大小并格式化显示

```htmlPython 计算文件大小并格式化显示

Python 计算文件大小并格式化显示

在处理文件时，我们经常需要知道文件的大小。Python 提供了多种方法来计算文件大小，并且可以将结果以易于理解的格式显示出来。本文将介绍如何使用 Python 来计算文件大小，并将其格式化为人类可读的形式。

计算文件大小

要计算文件大小，我们可以使用内置的 os 模块中的 stat 方法。这个方法返回一个包含文件信息的元组，其中 st_size 属性表示文件的字节数。

        import os

        def get_file_size(file_path):
            return os.stat(file_path).st_size
    

上述代码定义了一个函数 get_file_size，它接受一个文件路径作为参数，并返回该文件的大小（以字节为单位）。

格式化文件大小

虽然直接获取到的文件大小是以字节为单位的，但在实际应用中，我们更希望看到的是 KB、MB 或 GB 等形式的大小。下面是一个将字节数转换为人类可读格式的函数：

        def format_size(size_bytes):
            if size_bytes == 0:
                return "0B"
            size_name = ("B", "KB", "MB", "GB", "TB", "PB", "EB", "ZB", "YB")
            i = int(math.floor(math.log(size_bytes, 1024)))
            p = math.pow(1024, i)
            s = round(size_bytes / p, 2)
            return "%s %s" % (s, size_name[i])
    

这个函数首先检查文件大小是否为零，如果是，则直接返回 "0B"。然后，它定义了一个包含不同单位的元组 size_name。接着，通过计算对数和幂运算，确定应该使用的单位，并进行相应的转换和四舍五入，最后返回格式化的字符串。

完整示例

现在，让我们将这两个函数结合起来，创建一个完整的示例，用于计算并格式化显示文件大小：

        import os
        import math

        def get_file_size(file_path):
            return os.stat(file_path).st_size

        def format_size(size_bytes):
            if size_bytes == 0:
                return "0B"
            size_name = ("B", "KB", "MB", "GB", "TB", "PB", "EB", "ZB", "YB")
            i = int(math.floor(math.log(size_bytes, 1024)))
            p = math.pow(1024, i)
            s = round(size_bytes / p, 2)
            return "%s %s" % (s, size_name[i])

        file_path = "/path/to/your/file.txt"
        size_bytes = get_file_size(file_path)
        formatted_size = format_size(size_bytes)
        print("File size:", formatted_size)
    

在这个示例中，我们首先导入了 os 和 math 模块，然后定义了 get_file_size 和 format_size 函数。接下来，我们指定了一个文件路径，并调用这两个函数来计算和格式化文件大小，最后打印出结果。

总结

通过本文的介绍，我们学会了如何使用 Python 计算文件大小，并将其格式化为人类可读的形式。这种方法不仅可以帮助我们更好地理解和管理文件，还可以在开发过程中提供有用的调试信息。

```