如何添加或增加字典条目？

Question

我现在正在重新接触 Python，因为我很喜欢它。但是，我发现自己一遍又一遍地遇到一种模式。我一直在想，一定有更好的方式来表达我想要的东西，而我可能做错了。

我正在编写的代码格式如下：

# foo is a dictionary
if foo.has_key(bar):
  foo[bar] += 1
else:
  foo[bar] = 1

我在我的程序中写了很多。我的第一反应是把它推到一个辅助函数中，但是 python 库经常提供这样的东西。

我缺少一些简单的小语法技巧吗？还是应该这样做？

Answer 1

我不知道这是如何尝试的，但是如果您需要在 dict 键中附加项目...

indicatorDict = {}
indicatorDict[0] = 'Langford'
indicatorDict[1] = 'Esther'
indicatorDict[3] = 14

向其附加项目，无论是迭代还是其他类型：

indicatorDict[0] = np.append(indicatorDict[0],'Auditorium')
indicatorDict[1] = np.append(indicatorDict[1],'Duflo')
indicatorDict[3] = np.append(indicatorDict[3],'November') 

打印...

{0: array(['Langford', 'Auditorium'], dtype='<U10'),
 1: array(['Esther', 'Duflo'], dtype='<U6'),
 3: array(['14', 'November'], dtype='<U11')}

我避免了 Dict 中的第三个键，以表明如果需要，可以将键从一个步骤跳到另一个步骤... :) 希望对您有所帮助！

Answer 2

我做了一些时间比较。几乎相等。不过，单行的.get() 命令是最快的。

输出：

get 0.543551800627
exception 0.587318710994
haskey 0.598421703081

代码：

import timeit
import random

RANDLIST = [random.randint(0, 1000) for i in range(10000)]

def get():
    foo = {}
    for bar in RANDLIST:
        foo[bar] = foo.get(bar, 0) + 1


def exception():
    foo = {}
    for bar in RANDLIST:
        try:
            foo[bar] += 1
        except KeyError:
            foo[bar] = 1


def haskey():
    foo = {}
    for bar in RANDLIST:
        if foo.has_key(bar):
            foo[bar] += 1
        else:
            foo[bar] = 1


def main():
    print 'get', timeit.timeit('get()', 'from __main__ import get', number=100)
    print 'exception', timeit.timeit('exception()', 'from __main__ import exception', number=100)
    print 'haskey', timeit.timeit('haskey()', 'from __main__ import haskey', number=100)


if __name__ == '__main__':
    main()

Answer 3

dict 的get() 方法采用可选的第二个参数，如果找不到请求的键，可以使用该参数提供默认值：

foo[bar] = foo.get(bar, 0) + 1

Answer 4

使用defaultdict：

from collections import defaultdict

foo = defaultdict(int)
foo[bar] += 1

在 Python >= 2.7 中，您还有一个单独的 Counter 类用于这些目的。对于 Python 2.5 和 2.6，您可以使用其backported version。

Answer 5

您还可以利用异常处理中的控制结构。当您尝试将值分配给不存在的键时，字典会抛出 KeyError 异常：

my_dict = {}
try:
    my_dict['a'] += 1
except KeyError, err:    # in 2.6: `except KeyError as err:`
    my_dict['a'] = 1

Answer 6

对于 Python >= 2.5，您可以执行以下操作：

foo[bar] = 1 if bar not in foo else foo[bar]+1