根据两个列表中元组的出现更新 defaultdict 计数

Question

我有两个列表，想创建一个字典来记录元组的出现。

我当前的代码：

tup_to_find_test = [('good', 'pea'), ('leaf', 'sweet')] 
self_per_list_test = [('leaf', 'liquid'), ('leaf', 'sweet'), ('leaf', 'sweet'),('good', 'pea'),('good', 'pea'),('good', 'pea')]
from collections import defaultdict
tup_dict_test = defaultdict(int)
for tup_to_find_test in self_per_list_test:
   tup_dict_test[tup_to_find_test]+=1

我的结果是：

defaultdict(int, {('leaf', 'liquid'): 1, ('leaf', 'sweet'): 1, ('good', 'pea'): 3})

我想要的结果是：

('leaf', 'liquid'): 0, ('leaf', 'sweet'): 2, ('good', 'pea'): 3})

不知道为什么('leaf', 'liquid')的计数是1。defaultdict(int)的默认整数不就是零吗？为什么('leaf', 'liquid') tuple 得到 1？

Answer 1

这条线并没有像你想象的那样做：

for tup_to_find_test in self_per_list_test:
   # ...

您在此处逐元素迭代列表，在本例中为self_per_list_test 的元素。 没有过滤。当您的for 循环迭代时，tup_to_find_test 依次代表('leaf', 'liquid')、('leaf', 'sweet') 等。名称与您之前定义的变量相同的事实只会造成混淆。

相反，您可以使用三元语句来区分操作：

for item in self_per_list_test:
    tup_dict_test[item] += 1 if item in tup_to_find_test else 0

print(tup_dict_test)

defaultdict(int, {('leaf', 'liquid'): 0, ('leaf', 'sweet'): 2, ('good', 'pea'): 3})

---

collections.Counter 更符合 Python 的习惯。在字典理解中使用 set 进行 O(1) 查找是一种很好的做法。

from collections import Counter

tup_to_find_set = set(tup_to_find_test)
counts = Counter(self_per_list_test)

tup_dict_test = {k: v if k in tup_to_find_set else 0 for k, v in counts.items()}

print(tup_dict_test)

{('leaf', 'liquid'): 0, ('leaf', 'sweet'): 2, ('good', 'pea'): 3}

Answer 2

defaultdict(int)的默认整数不是零吗？

是的。

为什么我的 ('leaf', 'liquid') 元组得到 1？

你写道：

tup_dict_test[tup_to_find_test]+=1

也就是说，找到当前值（创建一个设置为零的新值），然后将其加一并将结果存储回来。结果值为 1。

Answer 3

无需重新发明轮子。为此，您可以使用美妙的collections 标准模块库中的counter。

from collections import Counter

tup_to_find_test = [('good', 'pea'), ('leaf', 'sweet')] 
self_per_list_test = [('leaf', 'liquid'), ('leaf', 'sweet'), ('leaf', 'sweet'),('good', 'pea'),('good', 'pea'),('good', 'pea')]

c = Counter(self_per_list_test)

for key in c:
    if key not in tup_to_find_test:
        c[key] = 0

print(c)

>>Counter({('good', 'pea'): 3, ('leaf', 'sweet'): 2, ('leaf', 'liquid'): 0})

这里我们基于self_per_list_test 创建一个计数器，如果在tup_to_find_test 中找不到它，则将计数更新为零。希望这是解决您问题的更直观的方法。