构建字典时增加值的Pythonic方法[重复]

Question

这个问题已经在这里有了答案：

• 检查给定键是否已存在于字典中并递增 12个答案

工作脚本中的一段代码； 我只是好奇是否有一种“更漂亮”的方式来达到相同的结果。

    if ctry in countries:
        countries[ ctry ] += 1
    else:
        countries[ ctry ] = 1

在awk中，我本可以使用countries[ ctry ] += 1 ，但是python抛出了一个关键错误（可以理解）。

Answer 1

您可以将countries更改为collections.defaultdict对象，而不是使用常规词典。 顾名思义，如果键尚不存在， collections.defaultdict允许您在字典中插入默认值：

from collections import defaultdict
countries = defaultdict(int)

然后，您的代码段变成一行：

countries[cntry] += 1

如果您不能使用collections.defaultdict ，则可以使用“请求宽恕而不是许可”的惯用语：

try:
    countries[ ctry ] += 1
except KeyError:
    countries[ ctry ] = 1

尽管上面的行为类似于您的条件语句，但是它被认为更“ Pythonic”，因为使用了try/except而不是if/else 。

Answer 2

以下是更多的pythonic：

countries[ctry]  = 1 if ctry not in countries else countries[ctry] + 1

要么

countries[ctry] = countries.get(ctry, 0) + 1

Answer 3

另一种选择是使用defaultdict：

from collections import defaultdict
countries = defaultdict(int)
countries[ctry] += 1

速度测试：

%timeit countries['Westeros'] += 1
10000000 loops, best of 3: 79 ns per loop

countries = {}
%timeit countries['Westeros'] = countries.get('Westeros', 0) + 1
1000000 loops, best of 3: 164 ns per loop