Question

我在python中创建dict，这个dict可能包括list和sub dicts。如果我想检查任何字典中存在的密钥，我首先达到该字典，然后只有我可以检查密钥是否存在。但有没有办法我可以使用父dict检查密钥存在只意味着如果父dict有任何子dicts然后搜索到每个级别让我知道密钥是否存在于整个dict中然后使用该密钥我可以插入或删除该dict中包含该密钥的元素。

例如，我有这个

[
    {
        'groupName': u'MainMenu_1',
        'menuItems': [
            {
                'action': u'/home\r\n',
                'sub_1': [

                ],
                'id': 1L,
                'name': u'Home\r\n'
            },
            {
                'action': u'/inventory',
                'sub_2': [

                ],
                'id': 2L,
                'name': u'Inventory\r\n'
            },
            {
                'action': u'/accounting\r\n',
                'sub_3': [
                    {
                        'action': u'/gl\r\n',
                        'name': u'GL\r\n',
                        'sub_4': [

                        ]
                    },
                    {
                        'action': u'/ap\r\n',
                        'name': u'AP\r\n',
                        'sub_5': [

                        ]
                    },
                    {
                        'action': u'/ar\r\n',
                        'sub_6': [

                        ],
                        'name': u'AR\r\n'
                    }
                ],
                'id': 3L,
                'name': u'Accounting\r\n'
            },
            {
                'action': u'/crm\r\n',
                'sub_8': [

                ],
                'id': 8L,
                'name': u'CRM\r\n'
            }
        ]
    },
    {
        'groupName': u'MainMenu_2',
        'menuItems': [
            {
                'action': u'/iv-receive\r\n',
                'sub_9': [

                ],
                'id': 9L,
                'name': u'Receiving\r\n'
            },
            {
                'action': u'/iv-shipping\r\n',
                'sub_10': [

                ],
                'id': 10L,
                'name': u'Shipping\r\n'
            }
        ]
    }
]

现在，如果在上面的示例中我想搜索任何键，例如sub_1，sub_3，sub_6，那么我将如何搜索此键

Answer 1

我们可以递归搜索所有符合条件的字典。以下实现将此类词典的所有引用附加到列表found：

def recursive_search(items, key):
    found = []
    for item in items:
        if isinstance(item, list):
            found += recursive_search(item, key)
        elif isinstance(item, dict):
            if key in item:
                found.append(item)
            found += recursive_search(item.values(), key)
    return found

found = recursive_search(items, 'sub_9')

Answer 2

def flatten(obj):
    if isinstance(obj, list):
        result = set()
        for e in obj:
            result = result.union(flatten(e))
        return result
    elif isinstance(obj, dict):
        return set(obj.keys()) | flatten(obj.values())
    else:
        return {obj}

您也可以将列表/词典展平为一组，并快速执行任意数量的搜索。

print(flatten(lst))
set([1, 2, 3, u'/inventory', 8, 9, 'menuItems', u'/home\r\n', u'/ar\r\n', u'AP\r\n', u'/accounting\r\n', 'id', u'MainMenu_2', u'Receiving\r\n', u'Home\r\n', u'MainMenu_1', u'/iv-shipping\r\n', 'groupName', 10, u'AR\r\n', u'Accounting\r\n', u'/ap\r\n', u'Inventory\r\n', u'CRM\r\n', u'/crm\r\n', u'/iv-receive\r\n', 'sub_10', u'GL\r\n', u'Shipping\r\n', 'name', 'sub_2', 'sub_3', 'sub_1', 'sub_6', 'sub_4', 'sub_5', 'action', 'sub_8', 'sub_9', u'/gl\r\n'])

检查整个列表中存在的键或python中的dict

2 个答案: