优化字典循环求和值

Question

我有一个采用嵌套字典input_dict的方法

final = 0
for key, value in input_dict[self.state][self.city].iteritems():
    age = self._get_age(key)
    if (age > 0 and age < MAX_VAL):
      final += value  * self.lookup[key][age] * self.multiplier

return final

这在大约 0.03 秒内运行，但考虑到在示例执行中它被调用 >10k 次，它最终成为瓶颈并负责大约 50% 的运行时间。假设我不能减少调用该方法的总次数，有没有人有关于如何改进这个的建议？

Answer 1

也许考虑类似以下的事情-

current_period = self.current_period - (self.current_period % 7)
MIN_VALUE = current_period - 7 * MAX_VALUE
return self.multiplier * sum(value * self.lookup[key][self._get_age(key)]
    for key, value in input_dict[self.state][self.city].iteritems()
    if MIN_VALUE < key < current_period
)

在这里，我将乘法 self.multiplier 拉出循环，并将比较 0 < age < MAX_VALUE 替换为预先计算值的等效比较，通过将 age 替换为 cmets 中描述的 _get_age() 方法获得并求解为key。这允许我们在age <= 0 or age >= MAX_VALUE 的情况下跳过函数调用+额外的计算，并且不会在原始0 < age < MAX_VALUE 上产生额外的成本（除了计算循环外的2个变量）。此外，这允许我们使用内置的 sum() 函数，该函数通常比通过 for 循环求和更快，但无需像 qxz 的回答那样创建单独的生成器。

请注意，我假设您的_get_age() 方法中的(self.current_period - period) 是一个整数，因此/ 7 将结果置于Python-2.x 中。如果不是这种情况，请从 current_period 分配中删除 - (self.current_period % 7) 以获得等效功能。

Answer 2

built-in sum function 通常比写出for 循环更快。 （请参阅this question。）在您的情况下，您可以构造要求和的值的generator expression，然后将其传递给sum：

items = (
    (key,value,self._get_age(key))
    for key,value in input_dict[self.state][self.city].iteritems()
)
return sum(
    value * self.lookup[key][age] * self.multiplier
    for key,value,age in items
    if 0 < age < MAX_VAL
)