如何执行此操作-python字典遍历和搜索

Question

我有嵌套字典：

{'key0': {'attrs': {'entity': 'p', 'hash': '34nj3h43b4n3', 'id': '4130'},
          u'key1': {'attrs': {'entity': 'r',
                              'hash': '34njasd3h43b4n3',
                              'id': '4130-1'},
                    u'key2': {'attrs': {'entity': 'c',
                                        'hash': '34njasd3h43bdsfsd4n3',
                                        'id': '4130-1-1'}}},
          u'key3': {'attrs': {'entity': 'r',
                              'hash': '34njasasasd3h43b4n3',
                              'id': '4130-2'},
                    u'key4': {'attrs': {'entity': 'c',
                                        'hash': '34njawersd3h43bdsfsd4n3',
                                        'id': '4130-2-1'}},
                    u'key5': {'attrs': {'entity': 'c',
                                        'hash': '34njawersd3h43bdsfsd4n3',
                                        'id': '4130-2-2'}}}},
 'someohterthing': 'someothervalue',
 'something': 'somevalue'}

给定一个id-所有ids中的一个，比如4130到4130-2-2。
找到正确的字典最简单的方法是什么？

就像如果给定的id是4130-2-1，那么它应该使用key=key5到达字典

请使用非xml方法。

编辑（1）：嵌套在1到4级别之间，但在解析之前我知道嵌套。

编辑（2）：修复了代码。

**编辑（3）：**再次修复了ids字符串值的代码。请原谅造成的混乱。我希望这是最后一次：）

Answer 1

如果要以一般方式解决问题，不管dict中有多少级别的嵌套，请创建一个递归函数，该函数将遍历树：

def traverse_tree(dictionary, id=None):
    for key, value in dictionary.items():
        if key == 'id':
            if value == id:
                print dictionary
        else:
             traverse_tree(value, id)
    return

>>> traverse_tree({1: {'id': 2}, 2: {'id': 3}}, id=2)
{'id': 2}

Answer 2

你的结构很不规则。这是一个带有Visitor函数的版本，该函数遍历attrs子词典。

def walkDict( aDict, visitor, path=() ):
    for  k in aDict:
        if k == 'attrs':
            visitor( path, aDict[k] )
        elif type(aDict[k]) != dict:
            pass
        else:
            walkDict( aDict[k], visitor, path+(k,) )

def printMe( path, element ):
    print path, element

def filterFor( path, element ):
    if element['id'] == '4130-2-2':
        print path, element

你会这样用的。

walkDict( myDict, filterFor )

它可以变成一个生成器，而不是一个访问者；它将yield path, aDict[k]而不是调用访问者函数。

你可以在for循环中使用它。

for path, attrDict in walkDictIter( aDict ):
    # process attrDict...

Answer 3

使用正确的类定义（而不是泛型字典）通常可以更好地解决此类问题。

class ProperObject( object ):
    """A proper class definition for each "attr" dictionary."""
    def __init__( self, path, attrDict ):
        self.path= path
        self.__dict__.update( attrDict )
    def __str__( self ):
        return "path %r, entity %r, hash %r, id %r" % (
            self.path, self.entity, self.hash, self.id )

masterDict= {} 
def builder( path, element ):
    masterDict[path]= ProperObject( path, element )

# Use the Visitor to build ProperObjects for each "attr"
walkDict( myDict, builder )

# Now that we have a simple dictionary of Proper Objects, things are simple
for k,v in masterDict.items():
    if v.id == '4130-2-2':
        print v

另外，既然已经有了正确的对象定义，可以执行以下操作

# Create an "index" of your ProperObjects
import collections
byId= collections.defaultdict(list)
for k in masterDict:
    byId[masterDict[k].id].append( masterDict[k] )

# Look up a particular item in the index
print map( str, byId['4130-2-2'] )