python dict：获取vs setdefault

Question

以下两个表达式似乎与我等效。哪一个更好？

data = [('a', 1), ('b', 1), ('b', 2)]

d1 = {}
d2 = {}

for key, val in data:
    # variant 1)
    d1[key] = d1.get(key, []) + [val]
    # variant 2)
    d2.setdefault(key, []).append(val)

结果是相同的，但是哪个版本更好或更确切地说是pythonic？

我个人觉得版本2更难理解，对我而言setdefault非常难以掌握。如果我理解正确，它将在字典中查找“键”的值（如果不可用），在字典中输入“
[]”，返回对该值或“ []”的引用，并在该值后附加“ val”参考。虽然肯定很平滑，但至少（至少对我而言）不是直观的。

在我看来，版本1更容易理解（如果可用，请获取“ key”的值，否则请获取“ []”，然后加入由[val]组成的列表并将结果放入“ key”
）。但是，虽然更直观易懂，但我担心此版本的性能会下降，所有这些列表都会创建。另一个缺点是“
d1”在表达式中出现两次，这很容易出错。可能有一个使用get的更好的实现，但是目前我无法理解。

我的猜测是，尽管经验不足的人更难以掌握版本2，但它会更快，因此更可取。意见？

Answer 1

你的两个例子做同样的事情，但并不意味着get与setdefault做。

两者之间的区别基本上是手动设置d[key]为每次指向列表，而不是仅在未设置时才setdefault自动设置d[key]为列表。

我使这两种方法尽可能相似

from timeit import timeit

print timeit("c = d.get(0, []); c.extend([1]); d[0] = c", "d = {1: []}", number = 1000000)
print timeit("c = d.get(1, []); c.extend([1]); d[0] = c", "d = {1: []}", number = 1000000)
print timeit("d.setdefault(0, []).extend([1])", "d = {1: []}", number = 1000000)
print timeit("d.setdefault(1, []).extend([1])", "d = {1: []}", number = 1000000)

并得到

0.794723378711
0.811882272256
0.724429205999
0.722129751973

因此setdefault比get为此目的快10％。

该get方法可以使您做的事 少于
使用setdefault。KeyError即使您不想设置密钥，也可以使用它来避免在密钥不存在时得到提示（如果那会经常发生）。

请参阅“ setdefault”的dict方法的用例，并且dict.get（）方法返回一个指针，以获取有关这两种方法的更多信息。

关于主题的setdefault结论是，大多数情况下，您想使用defaultdict。关于该线程的get结论是它很慢，通常最好（在速度方面）使用defaultdict或处理错误（取决于字典的大小和您的用例）进行双重查找。