如何使用减少字典

Question

我有一些问题需要了解如何在python中使用reduce和词典。例如，我有以下字典。

{1: 3, 2: 1, 3: 2}

我正在尝试计算以下内容：

s = 0
for i in h:
    s += h[i] * (h[i] - 1)

这按预期工作（我得到：8），但我尝试将其转换为减少范例失败： reduce(lambda x, y: x + y * (y - 1), h)，但我得错了答案。

我认为这是因为我使用的是键，而不是值。如何将我的代码转换为正确缩小？

Answer 1

您需要迭代字典，同时使用初始值零来减少字典。

注意，迭代字典，实际上遍历键，因此您需要索引字典以获取值

reduce(lambda x, key:x + h[key] * (h[key] - 1), h, 0)

或者，因为您只对字典的值感兴趣，至少关心密钥，只需迭代字典的值

Python 2.X

reduce(lambda x, value:x + value * (value - 1), h.itervalues(), 0)

Python 3.X

reduce(lambda x, value:x + value * (value - 1), h.values(), 0)