从可能缺少的dict值初始化Python变量

时间:2011-05-30 23:16:26

标签: python dictionary initialization indexing

Pythonic的做法是什么?

value = (mydict.has_key('index') and mydict['index']) or 1024

从dict的值初始化(如果存在),否则使用默认值。

在LBYL和EAFP上得到了一些答案,但代码太详细了:p

我想要一些单线初始化。

PS:仅限python-2.4(运行CentOS5)

4 个答案:

答案 0 :(得分:11)

value = mydict.get('index', 1024)

(请注意,这与您的代码并不完全相同,因为仅当密钥1024不存在时它才会使用index,而代码在值时也使用1024对应于键index是假的。从你的解释我推断前者是你真正想要的。)

答案 1 :(得分:2)

或者使用setdefault() - 如get(),但在返回值之前插入密钥:

value = mydict.setdefault('index', 1024)

答案 2 :(得分:1)

getsetdefault方法非常重要,它们应该位于任何Python程序员的工具箱中。

这样做的另一个重要方法是使用collections.defaultdict,因为这样您就可以在一个地方定义“默认”,而无需在使用字典的任何地方将其分散,即不要重复你自己。然后,如果您更改默认值,则无需通过代码查找所有这些获取和setdefaults。例如

>>> import collections
>>> myDict = collections.defaultdict
>>> myDict = collections.defaultdict(lambda : 1024)
>>> myDict[0] += 1
>>> print myDict[0], myDict[100]
1025 1024

因为您可以设置函数调用,所以您可以为默认值执行任何操作。如何为每个新密钥分配1到10之间的随机值?

>>> import collections
>>> import random
>>> myDict = collections.defaultdict(lambda : random.randint(1,10))
>>> print myDict[0], myDict[100]
1 5
>>> print myDict[0], myDict[100], myDict[2]
1 5 6

好的,这个例子有点人为,但我相信你能想到更好的用法。假设您要构建一个昂贵的类 - 只有在实际不存在键时才会创建实例,这与调用getsetdefault时不同,例如。

>>> class ExpensiveClass(object):
>>>     numObjects = 0
>>>     def __init__(self):
>>>         ExpensiveClass.numObjects += 1
>>> 
>>> print ExpensiveClass.numObjects
0
>>> x = {}
>>> x[0] = ExpensiveClass()
>>> print ExpensiveClass.numObjects
1
>>> print x.get(0, ExpensiveClass())
<__main__.ExpensiveClass object at 0x025665B0>
>>> print ExpensiveClass.numObjects
2
>>> ExpensiveClass.numObjects = 0
>>> z = collections.defaultdict(ExpensiveClass)
>>> print ExpensiveClass.numObjects
0
>>> print z[0]
>>> <__main__.ExpensiveClass object at 0x02566510>
>>> print ExpensiveClass.numObjects
1

请注意,对x.get的调用会构造一个新的ExpensiveClass实例,即使我们之前没有使用该实例,因为x[0]已经存在。如果您确实尝试计算ExpensiveClass个对象的数量,或者创建它们的速度很慢,则可能会出现问题。 collections.defaultdict没有出现这些问题。

答案 3 :(得分:1)

collections模块defaultdict类可能会有所帮助。不幸的是,它直到Python 2.5才被引入,并且它没有完全你的代码所做的事情。这里有一些不同之处:

  • 它仅在缺少某个密钥时提供一个值,而不是在该密钥存在但是具有值为False的值时提供。
  • 如果缺少查找键,它会将其添加到字典中并使用默认值并返回默认值。
  • 所有缺失值都会获得相同的默认值。

如果没关系,你可以用这样的东西模仿老蟒蛇中的一个的本质:

try:
    from collections import defaultdict
except ImportError:
    class defaultdict(dict):
        def __init__(self, default_factory=None, *a, **kw):
            dict.__init__(self, *a, **kw)
            self.default_factory = default_factory

        def __getitem__(self, key):
            try:
                return dict.__getitem__(self, key)
            except KeyError:
                return self.__missing__(key)

        def __missing__(self, key):
            self[key] = value = self.default_factory()
            return value

ActiveState提供了更完整仿真的配方。即使您使用的是Python 2.5+,也可能需要派生自己的类进行自定义以获得所需的确切行为。除了检查“False”真值并且可能没有向字典中添加任何缺失值之外,您还可以为其提供与不同键对应的多个默认值的单独字典,而不是“一个尺寸适合所有”。有关方法,请参阅Is there a way to set multiple defaults on a Python dict using another dict?