当更新依赖于值本身时,是否有一种 pythonic 方法来更新字典值



我发现自己在很多情况下都有一个字典值,我想用一个新值更新,但前提是新值满足相对于当前值的一些条件(例如更大)。

目前,我编写的表达式类似于:

dictionary[key] = max(newvalue, dictionary[key])

这工作正常,但我一直在想,可能有一种更整洁的方法可以做到这一点,而不涉及重复自己。

感谢您的任何建议。

您可以使用封装该逻辑的更新方法使值对象。 或者子类字典并修改__setitem__的行为。 请记住,您所做的任何类似操作都会使不熟悉您的代码的人不太清楚发生了什么。 你现在正在做的事情是最明确和清楚的。

只需为自己编写一个辅助函数:

def update(dictionary, key, newvalue, func=max):
    dictionary[key] = func(dictionary[key], newvalue)

不确定它是否"更整洁",但避免重复自己的一种方法是使用面向对象的方法并对内置的dict类进行子类类化,以使某些东西能够做你想做的事情。这样做还有一个优点,即可以使用自定义类的实例代替dict实例,而无需更改代码的其余部分。

class CmpValDict(dict):
    """ dict subclass that stores values associated with each key based
       on the return value of a function which allow the value passed to be
       first compared to any already there (if there is no pre-existing
       value, the second argument passed to the function will be None)
    """
    def __init__(self, cmp=None, *args, **kwargs):
        self.cmp = cmp if cmp else lambda nv,cv: nv  # default returns new value
        super(CmpValDict, self).__init__(*args, **kwargs)
    def __setitem__(self, key, value):
        super(CmpValDict, self).__setitem__(key, self.cmp(value, self.get(key)))
cvdict = CmpValDict(cmp=max)
cvdict['a'] = 43
cvdict['a'] = 17
print cvdict['a']  # 43
cvdict[43] = 'George Bush'
cvdict[43] = 'Al Gore'
print cvdict[43]  # George Bush

使用Python版本的三元运算符怎么样:

d[key]=newval if newval>d[key] else d[key]

或一行,如果:

if newval>d[key]: d[key]=newval

相关内容

最新更新