我发现自己在很多情况下都有一个字典值,我想用一个新值更新,但前提是新值满足相对于当前值的一些条件(例如更大)。
目前,我编写的表达式类似于:
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