【Python】多级目录下文件总数

最新推荐文章于 2023-03-16 11:07:22 发布

思考实践

最新推荐文章于 2023-03-16 11:07:22 发布

阅读量945

点赞数

分类专栏： Python 文章标签：遍历多级目录求文件总数

本文链接：https://blog.csdn.net/weixin_43332715/article/details/122090065

版权

Python 专栏收录该内容

74 篇文章 2 订阅

订阅专栏

比如统计数据集中图片数量（其他文件做法相同），实质上遍历多级目录

 #----------------------------------------------------#
 #   path是你的一级目录
 #   path -> second_category -> third_catecory
 #----------------------------------------------------#
def get_image_num(path, train_own_data):
    num = 0
    if train_own_data:
        train_path = os.path.join(path, 'second_catecory')
        for character in os.listdir(train_path):
            #----------------------------------------------------#
            #   这是二级目录的遍历
            #   os.listdir(character_path)遍历输入路径下有文件，返回一个列表，通过len()可以返回其数量
            #----------------------------------------------------#
            character_path = os.path.join(train_path, character)
            num += len(os.listdir(character_path))
    else:
        train_path = os.path.join(path, 'second_catecory')
        for alphabet in os.listdir(train_path):
            #-------------------------------------------------------------#
            #   这是三级目录的遍历，这里alphabet相当于third_category
            #-------------------------------------------------------------#
            alphabet_path = os.path.join(train_path, alphabet)
            for character in os.listdir(alphabet_path):
                #----------------------------------------------------#
                #   在大众类下遍历小种类。
                #----------------------------------------------------#
                character_path = os.path.join(alphabet_path, character)
                num += len(os.listdir(character_path))
    return num