Python：字典dict（关键字对应的值为list）

原创于 2021-05-19 09:44:21 发布 · 1.2k 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#dict #defaultdict #Python

Python库专栏收录该内容

56 篇文章

订阅专栏

本文介绍并对比了两种使用Python对文件进行分组的方法：一种是通过手动创建字典进行分组；另一种则是利用Python标准库collections中的defaultdict简化分组过程。通过实例展示了如何根据文件名前缀将图片文件归类。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

方法一

代码

    # method 1
    pages = {}
    page = []
    for img_name in os.listdir(args.image_dir):
        pre_str = img_name[:2]
        #print(pre_str)
        if pre_str not in pages.keys():
            tmp_list = []
            tmp_list.append(os.path.join(args.image_dir, img_name))
            pages[pre_str] = tmp_list
        else:
            pages[pre_str].append(os.path.join(args.image_dir, img_name))

    print(pages)

输出

在这里插入图片描述

方法二

代码

    # method 2
    from collections import defaultdict
    pages = defaultdict(list)
    for img_name in os.listdir(args.image_dir):
        pre_str = img_name[:2]
        #print(pre_str)
        pages[pre_str].append(os.path.join(args.image_dir, img_name))

    print(list(pages.items()))
    print(pages)