检查Python中列表或字典中是否存在键

Question

我在用Python创建一个字典，这个字典里面可能会有列表和子字典。如果我想检查某个键是否存在于字典中，我首先得找到那个字典，然后才能检查这个键是否存在。但是有没有什么方法可以只通过父字典来检查键的存在呢？也就是说，如果父字典有任何子字典，那么搜索可以深入到每一层，告诉我这个键是否在整个字典中存在。然后我可以用这个键来插入或删除包含这个键的元素。

举个例子，我有这个：

[
    {
        '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

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'])

你可以把列表或字典“压平”成一个集合，这样你就可以快速进行很多次搜索。

Answer 2

我们可以递归地搜索所有符合条件的字典。下面的实现会把所有符合条件的字典的引用添加到列表 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')