在Python中从字典中删除某些键的最快方法

2024-04-21

我正在寻找删除 python 字典中某些键的最快/有效的方法

这里有一些选项

for k in somedict.keys(): 
    if k.startswith("someprefix"): 
        del somedict[k]

or

dict((k, v) for (k, v) in somedict.iteritems() if not k.startswith('someprefix'))

从逻辑上讲,第一个片段在较小的字典上应该更快,它不会创建字典的副本,而是创建所有键的列表,但是双重查找和字典重建非常耗时。虽然第二个在更大的字典上速度更快,但需要两倍的内存。 我已经在一些小型基准测试中检查了我的假设。

有什么更快的吗?


不仅是del更容易理解,但似乎比pop() http://docs.python.org/library/stdtypes.html#dict.pop:

$ python -m timeit -s "d = {'f':1,'foo':2,'bar':3}" "for k in d.keys():" "  if k.startswith('f'):" "    del d[k]"
1000000 loops, best of 3: 0.733 usec per loop

$ python -m timeit -s "d = {'f':1,'foo':2,'bar':3}" "for k in d.keys():" "  if k.startswith('f'):" "    d.pop(k)"
1000000 loops, best of 3: 0.742 usec per loop

编辑:感谢 Alex Martelli 提供有关如何进行基准测试的说明。希望我没有在任何地方犯错。

首先测量复制所需的时间:

$ python -m timeit -s "d = {'f':1,'foo':2,'bar':3}" "d1 = d.copy()"
1000000 loops, best of 3: 0.278 usec per loop

复制字典的基准:

$ python -m timeit -s "d = {'f':1,'foo':2,'bar':3}" "d1 = d.copy()" "for k in d1.keys():" "  if k.startswith('f'):" "    del d1[k]"
100000 loops, best of 3: 1.95 usec per loop

$ python -m timeit -s "d = {'f':1,'foo':2,'bar':3}" "d1 = d.copy()" "for k in d1.keys():" "  if k.startswith('f'):" "    d1.pop(k)"
100000 loops, best of 3: 2.15 usec per loop

减去复制成本,我们得到 1.872 usecpop()1.672 为del.

本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)

在Python中从字典中删除某些键的最快方法 的相关文章

随机推荐