python - 如何将 dict.get() 的默认参数用于复杂数据

Question

这是一个使用 dict.get() 方法的默认参数来计算单词数的脚本:

word_list = ['abc', 'abc', 'def']
word_data = {}

for word in word_list:
    word_data[word] = word_data.get(word, 0) + 1

print(word_data)

这对于简单计数来说效果很好。但如果我想收集更多数据，比如 word_list 中单词条目的索引，那么我需要将这些额外数据插入到 get() 的默认参数和尾随表达式(“+ 1”)中。

我怎样才能做到这一点？

Answer 1

如果你想继续使用dict.get方法，那么，你可以这样做:

 for i, word in enumerate(word_list):
     elem = word_data.get(word, {'index':[], 'count':0})
     word_data.update({
         word: {
             'indexes': elem['indexes']+[i], 
             'count': elem['count']+1
         }
     })

然后，你会得到:

{'abc': {'count': 2, 'indexes': [0, 1]}, 'def': {'count': 1, 'indexes': [2]}}

尽管如此，在这种特殊情况下，也许只跟踪索引会更有效(因为每个单词出现的次数只是该列表的长度)，对吧？ :)