有没有一种从字典中删除键而不引发KeyError的单行方法?
if 'key' in my_dict:
del my_dict['key']
有没有一种从字典中删除键而不引发KeyError的单行方法?
if 'key' in my_dict:
del my_dict['key']
当前回答
我更喜欢不可变的版本
foo = {
1:1,
2:2,
3:3
}
removeKeys = [1,2]
def woKeys(dct, keyIter):
return {
k:v
for k,v in dct.items() if k not in keyIter
}
>>> print(woKeys(foo, removeKeys))
{3: 3}
>>> print(foo)
{1: 1, 2: 2, 3: 3}
其他回答
您可以使用字典理解创建一个新的字典,并删除该键:
>>> my_dict = {k: v for k, v in my_dict.items() if k != 'key'}
您可以按条件删除。如果密钥不存在,则无错误。
如果希望非常详细,可以使用异常处理:
try:
del dict[key]
except KeyError: pass
然而,如果键不存在,这比pop()方法慢。
my_dict.pop('key', None)
这对几个键来说并不重要,但如果你重复这样做,那么后一种方法是更好的选择。
最快的方法是:
if 'key' in dict:
del myDict['key']
但这种方法很危险,因为如果在两行之间删除“key”,就会引发KeyError。
如果您需要在一行代码中从字典中删除大量键,我认为使用map()非常简洁,而且Python可读:
myDict = {'a':1,'b':2,'c':3,'d':4}
map(myDict.pop, ['a','c']) # The list of keys to remove
>>> myDict
{'b': 2, 'd': 4}
如果需要在弹出字典中没有的值时捕获错误,请在map()中使用lambda,如下所示:
map(lambda x: myDict.pop(x,None), ['a', 'c', 'e'])
[1, 3, None] # pop returns
>>> myDict
{'b': 2, 'd': 4}
或者在python3中,必须使用列表理解:
[myDict.pop(x, None) for x in ['a', 'c', 'e']]
它起作用了。即使myDict没有'e'键,'e'也不会导致错误。
我花了一些时间才弄清楚my_dict.pop(“key”,“None”)到底在做什么。因此,我将添加以下内容作为答案,以节省其他人的搜索时间:
pop(键[,默认值])如果键在字典中,请删除它并返回其值,否则返回默认值。如果未给定默认值,且密钥不在字典,则会引发KeyError。
文档
我更喜欢不可变的版本
foo = {
1:1,
2:2,
3:3
}
removeKeys = [1,2]
def woKeys(dct, keyIter):
return {
k:v
for k,v in dct.items() if k not in keyIter
}
>>> print(woKeys(foo, removeKeys))
{3: 3}
>>> print(foo)
{1: 1, 2: 2, 3: 3}