在Python中初始化词典的最佳方法是什么?

时间:2009-03-16 19:22:52

标签: python autovivification

在Perl很多次,我会做这样的事情:

$myhash{foo}{bar}{baz} = 1

我如何将其翻译成Python?到目前为止,我有:

if not 'foo' in myhash:
    myhash['foo'] = {}
if not 'bar' in myhash['foo']:
    myhash['foo']['bar'] = {}
myhash['foo']['bar']['baz'] = 1

有更好的方法吗?

4 个答案:

答案 0 :(得分:98)

如果你需要的嵌套量是固定的,collections.defaultdict很棒。

e.g。嵌套两个深:

myhash = collections.defaultdict(dict)
myhash[1][2] = 3
myhash[1][3] = 13
myhash[2][4] = 9

如果你想进行另一种嵌套,你需要做类似的事情:

myhash = collections.defaultdict(lambda : collections.defaultdict(dict))
myhash[1][2][3] = 4
myhash[1][3][3] = 5
myhash[1][2]['test'] = 6

编辑:MizardX指出我们可以通过一个简单的函数获得完全的通用性:

import collections
def makehash():
    return collections.defaultdict(makehash)

现在我们可以做到:

myhash = makehash()
myhash[1][2] = 4
myhash[1][3] = 8
myhash[2][5][8] = 17
# etc

答案 1 :(得分:89)

class AutoVivification(dict):
    """Implementation of perl's autovivification feature."""
    def __getitem__(self, item):
        try:
            return dict.__getitem__(self, item)
        except KeyError:
            value = self[item] = type(self)()
            return value

测试:

a = AutoVivification()

a[1][2][3] = 4
a[1][3][3] = 5
a[1][2]['test'] = 6

print a

输出:

{1: {2: {'test': 6, 3: 4}, 3: {3: 5}}}

答案 2 :(得分:13)

是否有理由需要成为决定词?如果这个特定结构没有令人信服的理由,你可以简单地用一个元组索引dict:

mydict = {('foo', 'bar', 'baz'):1} # Initializes dict with a key/value pair
mydict[('foo', 'bar', 'baz')]      # Returns 1

mydict[('foo', 'unbar')] = 2       # Sets a value for a new key

如果使用元组键初​​始化dict,则需要使用括号,但在使用[]设置/获取值时可以省略括号:

mydict = {}                        # Initialized the dict
mydict['foo', 'bar', 'baz'] = 1    # Sets a value
mydict['foo', 'bar', 'baz']        # Returns 1

答案 3 :(得分:2)

我想直译将是:

 mydict = {'foo' : { 'bar' : { 'baz':1}}}

通话:

 >>> mydict['foo']['bar']['baz']

给你1。

但这对我来说看起来有点粗糙。

(我不是perl家伙,所以我猜你的perl会做什么)