繁体   English   中英

从 Python 字典中删除多个键

[英]Remove more than one key from Python dict

是否有任何有效的快捷方法可以一次从 python 字典中删除多个键?

例如;

x = {'a': 5, 'b': 2, 'c': 3}
x.pop('a', 'b')
print x
{'c': 3}

使用del语句

x = {'a': 5, 'b': 2, 'c': 3}
del x['a'], x['b']
print x
{'c': 3}

删除一些键

我测试了三种方法的性能:

d = dict.fromkeys('abcdefghijklmnopqrstuvwxyz')
remove_keys = set('abcdef')

# Method 1
for key in remove_keys:
    del d[key]

# Method 2
for key in remove_keys:
    d.pop(key)

# Method 3
{key: v for key, v in d.items() if key no in remove_keys}

以下是 1M 次迭代的结果:

  1. 1.88 秒 1.9 纳秒/迭代 (100%)
  2. 2.41 秒 2.4 纳秒/迭代 (128%)
  3. 4.15 秒 4.2 纳秒/迭代 (221%)

所以del是最快的。

安全地移除一些钥匙

但是,如果要安全删除 ,使其不会因 KeyError 而失败,则必须修改代码:

# Method 1
for key in remove_keys:
    if key in d:
        del d[key]

# Method 2
for key in remove_keys:
    d.pop(key, None)

# Method 3
{key: v for key, v in d.items() if key no in remove_keys}
  1. 2.03s 2.0 ns/iter (100%)
  2. 2.38 秒 2.4 纳秒/迭代 (117%)
  3. 4.11 秒 4.1 纳秒/迭代 (202%)

尽管如此, del是最快的。

我使用的一般形式是这样的:

  1. 生成要从映射中删除的键列表;
  2. 循环遍历列表并为每个调用del

例子:

假设我想删除映射中的所有字符串键。 生成它们的列表:

>>> x={'a':5,'b':2,'c':3,1:'abc',2:'efg',3:'xyz'}
>>> [k for k in x if type(k) == str]
['a', 'c', 'b']

现在我可以删除那些:

>>> for key in [k for k in x if type(k) == str]: del x[key]
>>> x
{1: 'abc', 2: 'efg', 3: 'xyz'}

暂无
暂无

声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM