说说在 Python 字典中如何在读取不存在的键时得到一个默认值

我就是我 2022-12-23 01:53 77阅读 0赞

如果有方法能够在 Python 字典类型中，当读取不存在的键时能够得到一个默认值，那么代码就会变得更加直观。通过 defaultdict 类型可以实现这个目的1。

我们来改写一个 “输出单词所在坐标” 的示例来说明使用 defaultdict 类型与使用 setDefault 方法之间的区别2。

改写后的完整示例代码如下：

import collections
    import logging
    
    logging.basicConfig(level=logging.DEBUG, format='%(levelname)s - %(message)s')
    
    '''
    defaultdict 示例
    :author:Deniro Lee
    '''
    
    import sys
    import re
    
    WORD_RE = re.compile(r'\w+')
    
    index = collections.defaultdict(list)
    with open(sys.argv[1], encoding='utf-8') as fp:
        for line_no, line in enumerate(fp, 1):
            for match in WORD_RE.finditer(line):
                word = match.group()
                column_no = match.start() + 1
                location = (line_no, column_no)
                index[word].append(location)
    
    for word in sorted(index, key=str.upper):
        logging.info('word -> %s；index[word] -> %s', word, index[word])

部分运行结果：

INFO - word -> a；index[word] -> [(1, 7), (1, 46), (1, 76), (1, 111), (3, 52), (3, 87), (19, 75), (19, 162)]
    INFO - word -> about；index[word] -> [(9, 83)]
    INFO - word -> acknowledged；index[word] -> [(1, 27)]

我们把这两种方法在示例中实际区别列举出来，这样看的更清楚。

使用 setDefault 方法：

index = {}
    ...
    index.setdefault(word, []).append(location)
    ...

使用 defaultdict 类型：

index = collections.defaultdict(list)
    index[word].append(location)

相对来说，defaultdict 类型的写法比使用 setDefault 方法的写法更加直观。

在 index\[word\] 表达式中，如果 word 在 index 中不存在，即字典的某个键不存在，那么会新建一个 list 作为 word 的值，并以 word 为键一并放入 index 字典中，相当于初始化操作。所以示例中可以直接把 location 对象append 到 index\[word\] 表达式中。

--------------------

参考资料：  
【1】Luciano Ramalho (作者)，安道，吴珂 (译者)．流畅的Python．人民邮电出版社，2017:143-145.  
【2】[说说在 Python 中如何使用 setDefault 方法提高效率][Python _ setDefault].

[Python _ setDefault]: https://deniro.blog.csdn.net/article/details/109702159