Question

增加存储在字典中的计数器的最快方法是什么？

因为我必须做几十万次同样的操作，所以我正在寻找比下面更有效的东西：

def funcA(a):
    keys = [x for x in range(1, 51)]
    adict = {key: 0 for key in keys}

    for k in adict.keys():  # this is the code I would like to improve
        if k <= a:
            adict[k] += 1
        else:
            break


import timeit
number = 100000

t1 = timeit.timeit(
        'funcA(5)',
        setup="from __main__ import funcA", number=number)
print(t1)

>>> 0.42629639082588255

尝试使用列表理解似乎会减慢一切，可能是因为它缺少break语句？

def funcB(a):
    # not working, invalid syntax
    keys = [x for x in range(1, 51)]
    adict = {key: 0 for key in keys}

    def _inc(x):
        x += 1
        return x

    [_inc(adict[k]) for k in adict.keys() if k <= a]

# Timing: 0.5831785711925477

注意：最初我有if float(k) <= float(a):，但由于我只期望数字（整数或浮点数），因此删除float()转换可以改善代码。这个假设合理吗？

注意2：正如几条评论中所述，break语句可能会在结果字典中产生意外结果，因此最好只做：

def funcA(a):
    keys = [x for x in range(1, 51)]
    adict = {key: 0 for key in keys}

    for k in adict:
        if k <= a:
            adict[k] += 1

# Timing: 0.5132114209700376

Answer 1

在你的情况下，你可以使用布尔（比较的结果）可以简单地转换为整数这一事实。它可能不是最快但它肯定是短的并且“相对”快：

def funcA(a):
     adict = {key: int(key <= a) for key in range(1, 51)}

这假设第二个函数实际上是你想要的，因为第一个函数可以因break而给出不同的结果。字典是无序的，因此无法为小于或等于a的键增加某些值。此外，它不会增加值，只是将它们设置为1或0，因为在这种情况下您实际上不需要添加。

然而，这不一定是最快的方式，因为它必须进行大量的函数调用和int查找。因此，我将按性能顺序（最快到最慢）呈现一些更等效的操作：

def cached_version():
    range_cache = range(1, 51)
    cache = dict.fromkeys(range_cache, 0)
    def inner(a):
        adict = cache.copy()
        for key in range_cache[:a]:  # requires a to be an integer!
            adict[key] = 1
        return adict
    return inner

func1 = cached_version()  # initialize cache

def func2(a):
    keys = range(1, 51)
    adict = dict.fromkeys(keys[:a], 1)   # requires a to be an integer!
    for key in keys[a:]:
        adict[key] = 0
    return adict

def func3(a):
    adict = {}
    for key in range(1, 51):
        if key <= a:
            adict[key] = 1
        else:
            adict[key] = 0
    return adict

def func4(a):
    return {key: 1 if key <= a else 0 for key in range(1, 51)}

def func5(a):
    keys = range(1, 51) 
    adict = dict.fromkeys(keys[:a], 1)  # requires a to be an integer!
    adict.update(dict.fromkeys(keys[a:], 0))
    return adict

def func6(a):
    return dict(zip(range(1, 51), [1]*a + [0]*(49-a)))  # requires a to be an integer!

from itertools import chain

def func7(a):
    return dict(zip(range(1, 51), chain([1]*a, [0]*(49-a))))  # requires a to be an integer!

def func8(a):  # the one I originally mentioned
     adict = {key: int(key <= a) for key in range(1, 51)}

时序是在Python 3.5，Windows 10上完成的，其他机器和其他Python版本可能存在差异。另请注意，如果您有更多键而不仅仅是range(1, 51)，则性能可能完全不同。

在字典中增加计数器的最快方法

1 个答案: