如何在 Python 中使用 NumPy 库对桑 shutil 中的文件夹结构进行统计分析?
import numpy as np
import shutil
# 获取桑 shutil 中的文件夹结构
folders = shutil.listdir(".")
# 创建 NumPy 数据框
data = np.zeros((len(folders), 3))
data["folder_name"] = folders
data["file_count"] = np.zeros(len(folders))
data["size"] = np.zeros(len(folders))
# 统计文件夹大小
for i, folder in enumerate(folders):
size = shutil.get_size(os.path.join(folder, "*.txt"))
data["size"][i] = size
# 打印结果
print(data)
注意:
-
shutil.listdir(".")
函数会返回包含所有文件和文件夹名的列表。 -
np.zeros((len(folders), 3))
创建一个包含 3 个列的 NumPy 数据框,其中每个列代表一个统计指标:文件夹名称、文件数量和文件大小。 -
shutil.get_size()
函数用于获取文件或文件夹大小。 -
os.path.join()
函数用于拼接路径。