遍历文件夹汇总为 df 表格,将每一层的文件夹/文件对应一列数据

读取文件目录汇总为表的代码如下

import os
import pandas as pd

def generate_table(root_dir):
    data = []
    for app_dir in os.listdir(root_dir):
        app_path = os.path.join(root_dir, app_dir)
        if not os.path.isdir(app_path):
            continue
    for page_dir in os.listdir(app_path):
        page_path = os.path.join(app_dir,page_dir)
        if not os.path.isdir(page_path):
            continue
        for file_name in os.listdir(page_path):
            if not file_name.endswith('.png'):
                continue
            data.append([app_dir,page_dir,file_name])
    df= pd.DataFrame(data, columns=['第一级目录','第二级目录','文件名']
    return df

root_dir = '/data/img'
df = generate_table(root_dir)
print(df)

基于汇总好的表格,统计不同第一级目录对应文件数量

result= df.groupby(['第一级目录']).agg({'文件名':'count'})
print(result)
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值