主页 / 问题 / 1498507
Accepted
Елена Сергеева
Asked:2023-02-21 15:49:48 +0000 UTC2023-02-21 15:49:48 +0000 UTC

在嵌套字典中查找元素的路径

  • 772

我有一个这样的嵌套字典：

{'a': {'s': {...}, 't': {...}}, 
 'b': {'q': {...}, 'y': {...}, 'k': {...}}, ['1', '2', '3'...], 
 'c': {'i': {{'p': [...], 'l': ['qw', 'rt']}}, 'm': {...}}}

您需要找到特定元素的路径。

例如，如果你想找到s = 'rt'，那么你应该得到['c', 'i', 'l']。

我试图做这样的事情，但到目前为止没有任何效果：

def find_elements(my_dict, st):
    for k, v in my_dict.items():
        if isinstance(v, dict):
            for k2, v2 in v.items():
                if isinstance(v2, list) and st in v2:
                    yield k
                    yield k2
                else:
                    find_elements(v2, st)
        elif isinstance(v, list):
            for l in v:
                if l == st:
                    yield k
        elif isinstance(v, str):
            if v == st:
                yield k

s = 'какая-то строка'
qw = []
for i in find_elements(my_dict, s):
    qw.append(i)
python

1 个回答

  1. Best Answer

    你可以递归地做，但你需要额外保存已经走过的路径，以免迷路。

    def find_elements(my_dict, st, path=[]):
    for k, v in my_dict.items():
        if v == st or st in v:
            return path + [k]
        elif isinstance(v, dict):
            if el := find_elements(v, st, path + [k]):
                return el


my_dict = {'a': {'s': {}, 't': {}}, 'b': {'q': {}, 'y': {}, 'k': {'1'}},
           'c': {'i': {'p': [], 'l': 'rt'}, 'm': {}}}

s = 'rt'
qw = find_elements(my_dict, s)
print(qw)

    结论：

    ['c', 'i', 'l']

    • 3

相关问题