【python】思维提升系列 - 如何实现文件的遍历

函数1：解压缩指定路径下的 ZIP 文件到指定目录

1. 导入Python标准库中的zipfile模块。

2. 定义函数 unzip_data，接收两个参数：src_path 表示源 ZIP 文件的路径，target_path 表示目标解压缩目录的路径。

3. 在函数内部：

   • 首先，检查目标目录是否存在，如果不存在，则执行以下步骤：
     • 使用 zipfile.ZipFile() 打开指定路径的 ZIP 文件，以只读模式。
     • 使用 extractall() 方法将 ZIP 文件中的所有内容解压缩到指定的目录 target_path 下。
     • 关闭 ZIP 文件。

4. 函数调用部分：

   • 调用 unzip_data 函数两次，分别解压缩了两个不同路径下的 ZIP 文件到相应的目录。

函数2：对给定目录进行文件类型统计及内存占用

1. 导入 Python 的标准库 os。

2. 定义两个空字典 size_dict 和 type_dict，用于分别存储文件大小和文件类型统计信息。

3. 定义函数 get_size_type，接收一个参数 path，表示指定目录的路径。

4. 在函数内部：

   • 使用 os.listdir(path) 获取指定目录下的所有文件和子目录列表。
   • 遍历列表中的每个文件和子目录：
     • 如果是子目录，则递归调用 get_size_type 函数，对其进行深度文件名解析。
     • 如果是文件，则获取其文件后缀名。
       • 如果文件没有后缀名，则在 type_dict 中以 “None” 作为键，记录文件数目，并在 size_dict 中记录文件大小。
       • 如果文件有后缀名，则在 type_dict 中以文件后缀名作为键，记录文件数目，并在 size_dict 中记录文件大小。

5. 函数执行完毕后，size_dict 中存储了不同文件类型的总内存占用情况，type_dict 中存储了不同文件类型的文件数目统计信息。

备注：通过调用 get_size_type 函数并传入指定目录的路径，可以统计该目录下所有不同子文件类型的文件数目和占用内存情况。

主函数调用：

1. 定义一个变量 path，表示指定目录的路径为 “data/”。

2. 调用函数 get_size_type(path)，获取指定目录下文件大小和类型的统计信息。

3. 遍历 type_dict 字典中的键，并针对每个文件类型执行以下操作：

   • 打印该文件类型所在路径、文件类型、文件数目和占用内存大小。

4. 打印总文件数，使用 sum() 函数对 type_dict 中的值进行求和。

5. 打印总内存大小，将 size_dict 中所有值相加，以 GB 为单位，保留两位小数。